python - 将点列表转换为 numpy 二维数组

Question

我正在使用 genfromtxt 导入一个二维数组，它的所有值都列在一个文本文件中，格式为(x 和 y 是整数):

    x1   y1   z1
    x2   y2   z2
    :    :    :

我正在使用下面的 for 循环，但我很确定必须有一种单行方法来完成它。执行此转换的更有效方法是什么？

raw = genfromtxt(file,skip_header = 6)

xrange = ( raw[:,0].min() , raw[:,0].max() )
yrange = ( raw[:,1].min() , raw[:,1].max() )

Z = zeros(( xrange[1] - xrange[0] +1 , yrange[1] - yrange[0] +1 ))

for row in raw:
    Z[ row[0]-xrange[0] , row[1]-yrange[0] ] = row[2]

Answer 1

您可以用以下代码替换 for 循环:

xidx = (raw[:,0]-xrange[0]).astype(int)
yidx = (raw[:,1]-yrange[0]).astype(int)

Z[xidx, yidx] = raw[:,2]