向Python numpy数组分配元素的问题

Posted 2023-01-20

我有一个3D数组，定义和初始化如下。我可以知道为什么以下作业不起作用吗？谢谢。

import numpy as np
xy = np.array([[(0,0) for _ in np.arange(0,2,0.5)] for _ in np.arange(0,2,0.5)])
xy[(0,0)] = (0,0.5)
print(xy[(0,0)])

输出：

[0 0]

所需的输出：

[0 0.5]

答案

请参阅下面的代码：

import numpy as np
xy = np.array([[(0,0) for _ in np.arange(0,2,0.5)] for _ in np.arange(0,2,0.5)])
xy = xy.view('float32')
xy[(0,0)] = (0,0.5)
print(xy[(0,0)])

输出

[0.  0.5]

另一答案

好的，这是一个具有正确的数组创建，赋值和变量名的版本：

import numpy as np

arr_1 = np.zeros(shape=(4, 4, 2))

arr_1[0, 0] = [0, 0.5]

print(arr_1[0, 0])的输出：[0. 0.5]