python - 使用 matplotlib 从 CSV 文件中绘制数据

Question

我有一个 CSV 文件，位于e:\dir1\datafile.csv. 它包含三列和 10 个标题和尾随行，需要跳过。我想用 numpy.loadtxt() 来绘制它，我还没有找到任何严格的文档。

这是我从网上找到的几次尝试中开始写的。

import matplotlib as mpl
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.cbook as cbook

def read_datafile(file_name):
    # the skiprows keyword is for heading, but I don't know if trailing lines
    # can be specified
    data = np.loadtxt(file_name, delimiter=',', skiprows=10)
    return data

data = read_datafile('e:\dir1\datafile.csv')

x = ???
y = ???

fig = plt.figure()

ax1 = fig.add_subplot(111)

ax1.set_title("Mains power stability")    
ax1.set_xlabel('time')
ax1.set_ylabel('Mains voltage')

ax1.plot(x,y, c='r', label='the data')

leg = ax1.legend()

plt.show()

Answer 1

根据文档 numpy.loadtxt是

用于简单格式文件的快速阅读器。genfromtxt 函数提供了更复杂的处理，例如缺失值的行。

所以只有几个选项可以处理更复杂的文件。如前所述numpy.genfromtxt，有更多选择。所以作为一个例子，你可以使用

import numpy as np
data = np.genfromtxt('e:\dir1\datafile.csv', delimiter=',', skip_header=10,
                     skip_footer=10, names=['x', 'y', 'z'])

读取数据并为列分配名称（或从文件中读取标题行names=True），然后用

ax1.plot(data['x'], data['y'], color='r', label='the data')

我认为 numpy 现在有很好的记录。ipython您可以从内部或使用 IDE轻松检查文档字符串，spider如果您更喜欢阅读呈现为 HTML 的文档字符串。

Answer 2

我正在猜测

x= data[:,0]
y= data[:,1]