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



我需要将 XYZ(int( 处的值(浮点数(从 CSV 文件插入到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 中,它拒绝同时处理两者,我也尝试过用熊猫但同样的问题。

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

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

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

改为执行以下操作:

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']

最新更新