在玩具示例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
有人知道我做错了什么吗?这使我很困惑,因为这种方法确实适用于玩具示例。
在此先感谢您,我希望问题已经足够清楚地阐明(这里是新手)。
答案 0 :(得分:1)
好的,因此在实际的dataFrame中,只选择了一部分数据进行绘图。 因此,子集的索引并非以0开头,这显然混淆了Python。 解决方案是使用以下方法重置索引:
df = df.reset_index(drop=True)
感谢帮助:)