python - Python：使用 3 列数据加载文本

Question

我有一个文本文件，其中包含要绘制的 3 列数据。

from numpy import *
import pylab
from mpl_toolkits.mplot3d import Axes3D

datalist = loadtxt("datagrid.txt")
x, t, u = datalist[:, 0, 0], datalist[0, :, 0], datalist[0, 0, :]

fig = pylab.figure()
ax = fig.add_subplot(111, projection = '3d')
ax.plot(x, t, u)
pylab.show()

我被告知太多的索引。如果我只能使用 2 个索引，如何解压缩具有 3 列的文本文件？

Answer 1

据我了解正确，该文件"datagrid.txt"包含类似

1 2 3
4 5 6
7 9 0
. . .
. . .
. . .

如果是这样，则将其loadtxt加载为二维数组。因此，您需要将行设置x、t和更改u为：

x, t, u = datalist[:,0], datalist[:,1], datalist[:,2]

甚至更简单、更 Pythonic：

x, t, u = datalist.transpose()

或作为第三个选项，将您的文件读取为：

x, t, u = loadtxt("datagrid.txt", unpack=True)