如何在python中简化数组

Question

当我将元素追加到空列表时，它往往会过于复杂。

我明白了：

A = array([[1],[1],...,[1]])

我想：

A = array([1,1...,1])

Answer 1

使用numpy.ndarray.flatten，

import numpy as np

A = np.array([[1], [1], [1]])
B = A.flatten()

Answer 2

你可以尝试将内循环附加到另一个列表：

A = ([[1],[4],[5]])
b = []
for x in A:
    for i in x:
     b.append(i)        
print(b)

输出：

[1, 4, 5]

Answer 3

要在向列表添加元素时避免这种情况，您可以使用extend代替append