Question

我有大量需要绘图的数据，这些数据作为xyz数据存储在3列中。我需要将列数据转换为网格，以便我可以在matplotlib中使用contourf轻松绘制它，我想知道是否有一个函数来执行此操作，因为我自己编写的代码非常慢？

即

到这样的网格：

10 12

14 16

Answer 1

numpy对此很聪明。您可以只读取不同数组中的列并执行：

import numpy

idx1 = numpy.array([0, 0, 1, 1])
idx2 = numpy.array([0, 1, 0, 1])
data = numpy.array([10, 12, 14, 16])

grid = numpy.zeros(len(data)/2, 2)
grid[idx1, idx2] = data

>>>grid
array([[ 10.,  12.],
      [ 14.,  16.]])

请记住，索引从0开始，所以如果从1开始，则需要从每个元素减1。

Answer 2

假设您的数据包含在data.txt中。以下代码将以正确的顺序打印出所需的数据部分。

假设data.txt在连续的行中有x和y坐标：

data.txt
x y z
1 1 10
1 2 12
2 1 14
2 2 16

def extract(filepath):
    f = open(filepath)
    f.readline() # to read out the first line that says "x y z"
    while 1:
        x = f.readline().strip().split()[-1]
        y = f.readline().strip().split()[-1]
        print x, y

请注意，当处理完文件的所有内容（但仍将打印所有值）时，这将以异常结束。为避免这种情况，请将f = open(filepath)替换为with open(filepath) as f:

但是，如果data.txt的结构不是那样，那么您需要利用每行中的前两个数字：

data.txt
x y z
1 1 10
2 1 14
1 2 12
2 2 16

from collections import defaultdict
def extract(filepath):
    coords = defaultdict(dict)
    f = open(filepath)
    f.readline() # to read out the first line that says "x y z"
    for line in f:
        id, axis, val = map(int, line.strip().split())
        coords[id][axis] = val

    for i in sorted(coords):
        print coords[i][1], coords[i][2]

希望这有帮助

列xyz数据到网格进行绘图

2 个答案: