0

在绘制 numpy.genfromtxt 命令返回的数据时，我无法弄清楚如何在 matplotlib 中使用“名称”。场景： 1. 我有一个包含列标题和值行的文件 2. 我事先不知道列标题——它们是通过程序生成的，并且可能在程序运行期间发生变化 3. 我需要读取数据和列标题，绘制它们并生成相应的图例。

我可以通过以下方式读取数据列及其名称：

dataArray = numpy.genfromtxt('myData.csv', delimiter = ',', names = True)

然后用

matplotlib.plot.plot(dataArray)
matplotlib.plot.show()

但是我如何制作一个合适的图例？我认为没有参数的图例命令就足够了（例如matplotlib.plot.legend()），但事实并非如此。我收到一个错误：

/usr/lib/python2.7/site-packages/matplotlib/axes.py:4601：用户警告：未找到标记的对象。在个别地块上使用 label='...' kwarg。warnings.warn("没有找到标记的对象。"

换句话说：那些“名字”去哪里了，我怎样才能找回它们？在 google、matplotlib 站点和 numy 站点上多次搜索没有产生任何结果。

4

1 回答 1

2

您必须在函数中为要绘制的每一行提供一个label=..关键字，因为 matplotlib 不会自动从 numpy 结构化数组中检测名称（您也可以使用 pandas，它会执行此操作，见下文）。plot

例如，您的数据如下所示：

from StringIO import StringIO

myDatacsv = StringIO("""a, b, c
1, 2, 3
2, 3, 4
3, 4, 5""")

读取它们numpy.genfromtxt会产生一个结构化数组：

>>> import numpy as np
>>> dataArray = np.genfromtxt(myDatacsv, delimiter = ',', names = True)
>>> dataArray
array([(1.0, 2.0, 3.0), (2.0, 3.0, 4.0), (3.0, 4.0, 5.0)], 
      dtype=[('a', '<f8'), ('b', '<f8'), ('c', '<f8')])

（在您的情况下，将有"myData.csv"而不是myDatacsv偏离路线，这只是为了举例说明）
现在您可以遍历列名并绘制它们中的每一个：

import matplotlib.pyplot as plt

plt.figure()
for col_name in dataArray.dtype.names:
    plt.plot(dataArray[col_name], label=col_name)

plt.legend()
plt.show()

这将生成如下图：

在此处输入图像描述

使用 pandas，这将产生相同的图形（自动绘制数据框的所有列并将其添加到图例中）：

import pandas as pd

# one of the following will do (reading it with pandas, or converting 
# from the numpy array to pandas dataframe)
data_df = pd.read_csv(myDatacsv)
data_df = pd.DataFrame(dataArray)

data_df.plot()

有关熊猫的更多信息，请参阅： http: //pandas.pydata.org/

于 2013-05-10T18:22:02.200 回答