创建具有不同元素大小的数组或列表python

Question

我想创建一个具有不同元素大小的数组或列表，但是，我得到的输出与在数组中间写入的单词数组不同，如下所示：

import numpy as np
lst_2=np.array([1,2,np.repeat(3,3),2],dtype="object")
print(lst_2) 
#Output is=[ 1 2 array([3,3,3]) 2]
#lst_2 should be = [1,2,3,3,3,2]...... 

请任何帮助或建议

Answer 1

您正在混合列表和 numpy 数组。 我假设您想要实现的是一个 numpy 数组，如下所示：

import numpy as np
lst_2=np.concatenate([np.array([1,2]), np.repeat(3,3),[2]])
print(lst_2) 

输出：

[1 2 3 3 3 2]

或者，如果需要列表，也可以使用以下内容：

lst_2 = [1,2] + list(np.repeat(3,3)) + [2]

这里的加号运算符代表两个列表的连接。 而在 numpy 数组的情况下，加号运算符代表逐元素加法。