python - 如何在 numpy 数组中的 XYZ 位置插入 csv 文件中的浮点值

Question

我需要将 CSV 文件中 XYZ(int) 处的值(Float)插入到 world = numpy.zeros((360,360,360))

CSV 文件中的行示例:

Value   X   Y   Z
4.5396  159 159 1
4.5905  182 160 3
etc

我的 CSV 文件中有大约 16000 行，尽管很多都是重复的。

我的主要问题是我需要使用 Integers(XYZ) 将 Float(Value) 插入 Numpy，它拒绝同时处理两者，我也尝试过使用 Pandas 但同样的问题。

在这里，我已将其全部转换为整数，这是我使其正常工作的唯一方法。

world = numpy.zeros((360,360,360))
data = numpy.genfromtxt('Myfile.csv', delimiter=',',skip_header=1) 
for Value,X, Y, Z in data.astype(int):
         world[X, Y, Z] = Value

它应该看起来像这样

world[159,159,1] = 4.5369
world[182,160,3] = 4.5945 

但我只能得到这个

world[159,159,1] = 4
world[182,160,3] = 4

其他一切都会导致各种错误

Answer 1

这样做:

world = numpy.zeros((360,360,360))
data = numpy.genfromtxt('Myfile.csv', delimiter=',', skip_header=1) 
X, Y, Z = data[:, 1:].astype(int).T
world[X, Y, Z] = data[:, 0]

或者直接使用正确的类型读取文本数据:

world = numpy.zeros((360,360,360))
data = numpy.genfromtxt('Myfile.csv', delimiter=',', skip_header=1,
                        dtype=[('Value', float), ('X', int), ('Y', int), ('Z', int)]) 
world[data['X'], data['Y'], data['Z']] = data['Value']