matplotlib:在数据框行上绘制一条迭代的行

时间:2018-11-01 10:40:48

标签: python pandas matplotlib

在玩具示例dataFrame中,有两组坐标:x,y和ex,ey。

d = {'x': [1, 2, 3, 4], 'y': [3, 3, 3, 3], 'ex': [1, 2, 3, 4], 'ey': [6, 6, 6, 6]}
toy = pd.DataFrame(data=d)

首先需要对每个集合进行散点图绘制,然后通过一条线进行连接。

plt.scatter(toy['x'],toy['y'], color='b')
plt.scatter(toy['ex'],toy['ey'], color='g')
plt.plot(toy['x'],toy['y'], color='b')
plt.plot(toy['ex'],toy['ey'], color='g')

最后,同一行中出现的集合之间的样本也必须通过线连接。通过将每一列作为pandas.Series类型来实现

x = toy['x']
ex = toy['ex']
y = toy['y']
ey = toy['ey']

并在绘图函数中对其进行迭代

for i in range(len(x)):
    plt.plot([x[i], ex[i]], [y[i], ey[i]], color='cyan')

它奏效了。

问题是,当采用实际的dataFrame时,此精确方法不起作用并返回以下错误:


KeyError                                  Traceback (most recent call last)
<ipython-input-174-aa1b4849722f> in <module>()
     21 
     22 for i in range(len(x)):
---> 23     plt.plot([x[i], ex[i]], [y[i], ey[i]], color='cyan')
     24 
     25 plt.show()

/usr/lib/python3/dist-packages/pandas/core/series.py in __getitem__(self, key)
    601         key = com._apply_if_callable(key, self)
    602         try:
--> 603             result = self.index.get_value(self, key)
    604 
    605             if not is_scalar(result):

/usr/lib/python3/dist-packages/pandas/indexes/base.py in get_value(self, series, key)
   2167         try:
   2168             return self._engine.get_value(s, k,
-> 2169                                           tz=getattr(series.dtype, 'tz', None))
   2170         except KeyError as e1:
   2171             if len(self) > 0 and self.inferred_type in ['integer', 'boolean']:

pandas/index.pyx in pandas.index.IndexEngine.get_value (pandas/index.c:3557)()

pandas/index.pyx in pandas.index.IndexEngine.get_value (pandas/index.c:3240)()

pandas/index.pyx in pandas.index.IndexEngine.get_loc (pandas/index.c:4279)()

pandas/src/hashtable_class_helper.pxi in pandas.hashtable.Int64HashTable.get_item (pandas/hashtable.c:8564)()

pandas/src/hashtable_class_helper.pxi in pandas.hashtable.Int64HashTable.get_item (pandas/hashtable.c:8508)()

KeyError: 0

有人知道我做错了什么吗?这使我很困惑,因为这种方法确实适用于玩具示例。

在此先感谢您,我希望问题已经足够清楚地阐明(这里是新手)。

1 个答案:

答案 0 :(得分:1)

好的,因此在实际的dataFrame中,只选择了一部分数据进行绘图。 因此,子集的索引并非以0开头,这显然混淆了Python。 解决方案是使用以下方法重置索引:

df = df.reset_index(drop=True)

感谢帮助:)