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] 

或者更簡單和更Python：

x, t, u = datalist.transpose() 

或第三個選項，閱讀您的文件爲：

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