“错误：使用序列设置数组元素”

Question

我正在尝试将 Matlab 代码转换为 Python，但是当我在数组中添加零时收到错误消息。

Matlab 代码：

N_bits=1e5;
a1=[0,1];
bits=a1(ceil(length(a1)*rand(1,N_bits)));
bits=[0 0 0 0 0 0 0 0 bits];

Python 代码：

a1=array([0,0,1])
N_bits=1e2
a2=arange(0,2,1)
## Transmitter ##
bits1=ceil(len(a2)*rand(N_bits))
bits=a1[array(bits1,dtype=int)]
bits=array([0,0,0,0,0,0,0,0, bits])

最后一行出现错误：

错误： 位=数组（[0,0,0,0,0,0,0,0，位]） ValueError：使用序列设置数组元素。

Answer 1

你想用数组加入列表，所以试试

bits=concatenate(([0,0,0,0,0,0,0,0], bits))

其中concatenate() 是numpy.concatenate()。您还可以使用zeros(8, dtype=int) 代替零列表（请参阅numpy.zeros()）。

与 Matlab 不同，Python 中的 [0,0,0,0,0,0,0,0, bits] 之类的东西会创建一个列表，其中初始零后跟 嵌入 列表。

Matlab：

>> x = [1,2,3]

x =

     1     2     3

>> [0,0,x]

ans =

     0     0     1     2     3

Python：

>>> x = [1,2,3]
>>>
>>> [0,0,x]
[0, 0, [1, 2, 3]]
>>> 
>>> [0,0] + x
[0, 0, 1, 2, 3]