在numpy数组中添加函数

问题描述 投票:-1回答:2

我的项目需要帮助。我有一个数组,看起来像这样?

rndm = [[0 1]
        [0 0]
        [0 0]
        [0 1]]

现在,我想添加 par_1 = [[1 0]], par_2 = [[0 0], ch1 = [[1 1]], and ch2 = [[0 1]] 到rndm。

我的代码是这样的。

new_rndm = []
new_rndm.append(par_1)
new_rndm.append(par_2)
new_rndm.append(ch1)
new_rndm.append(ch2)
# add them to rndm
rndm = numpy.append(rndm, [new_rndm])
print(rndm)

输出结果是这样的

rndm = [0 1 0 0 0 0 0 1 1 0 0 0 1 1 0 1]

我期待的输出结果是:

rndm = [[0 1]
        [0 0]
        [0 0]
        [0 1]
        [1 0]
        [0 0]
        [1 1]
        [0 1]]

我想问题在于append不能用于数组中。如果正确的话,有谁能帮助我,我可以尝试其他的函数吗?如果不对,请教我。我非常愿意学习。谢谢你!我的项目需要帮助。

python arrays python-3.x append add
2个回答
3
投票

使用 np.append(<array>, <elem to append>, axis=0)

rndm = np.array([[0, 1],
        [0, 0],
        [0, 0],
        [0, 1]])

par_1 = [[1, 0]]; par_2 = [[0, 0]]; ch1 = [[1, 1]]; ch2 = [[0, 1]]

rndm = np.append(rndm, par_1, axis=0)
rndm = np.append(rndm, par_2, axis=0)
rndm = np.append(rndm, ch1, axis=0)
rndm = np.append(rndm, ch2, axis=0)

array([[0, 1],
       [0, 0],
       [0, 0],
       [0, 1],
       [1, 0],
       [0, 0],
       [1, 1],
       [0, 1]])

编辑。

重塑。

x = np.array([2,1])
y = x.reshape(-1,1) # <------------ you have to do this
x.shape, y.shape

((2,), (2, 1))

0
投票

你可以用 .append 来添加一个数组到另一个数组的结尾。这里的问题是 numpy.append 先将数组平坦化,即 numpy.append([1 0], [0 1])[1 0 0 1]. 见 numpy docs on .append.


0
投票

您可以使用普通的列表追加来生成所需的嵌套列表结构。

rndm = [[0, 1],
        [0, 0],
        [0, 0],
        [0, 1]
        ]

par_1 = [[1, 0]]
par_2 = [[0, 0]]
ch1 = [[1, 1]]
ch2 = [[0, 1]]

new_rndm = []

new_rndm.append(par_1)
new_rndm.append(par_2)
new_rndm.append(ch1)
new_rndm.append(ch2)

new_rndm = [i for k in new_rndm for i in k]

for data in new_rndm:
    rndm.append(data)

for data in rndm:
    print(data)

输出:

[0, 1]
[0, 0]
[0, 0]
[0, 1]
[1, 0]
[0, 0]
[1, 1]
[0, 1]
© www.soinside.com 2019 - 2024. All rights reserved.