python - 使用 Seaborn 将散点图上对应的 x 和 y 事件用一条线连接起来

Question

我有一个包含 24 个事件的 csv 文件。第一列是“事件类型”，每个事件交替显示“开始”和“完成”，每个事件的 x 和 y 坐标都是随机的。这是 CSV 文件的样子。

CSV 文件

这是我使用此信息打印散点图的代码：

import seaborn as sns
Events=pd.read_csv('StartFinish.csv')
Scatter = sns.scatterplot(x='xCoordinate', y='yCoordinate', hue='Event Type', data=Events)

这是我在 Spyder 中运行此代码得到的结果：

我唯一的问题是我需要添加将每个“开始”事件与其相应的“完成”事件连接起来的行。（相应的“完成”事件是紧随其后发生的事件。）

除了 pandas、numpy、matplotlib.pyplot 和 seaborn 之外，我如何在不导入任何库的情况下执行此操作？

提前感谢您的帮助。

标签： pythonseaborn

Answer 1

import seaborn as sns
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
# Replace the following with your own dataframe
# Events=pd.read_csv('StartFinish.csv')
Events = np.random.randint(0, 10, size=(24, 2))
Events = pd.DataFrame(Events, columns=['xCoordinate', 'yCoordinate'])
Events['Event Type'] = 'Start' 
Events.loc[1::2, 'Event Type'] = 'Finish'

Scatter = sns.scatterplot(x='xCoordinate', y='yCoordinate', hue='Event Type', data=Events)
Scatter
for n in range(0, len(Events), 2):
    plt.plot([Events.loc[n, 'xCoordinate'], Events.loc[n+1, 'xCoordinate']], [Events.loc[n, 'yCoordinate'], Events.loc[n+1, 'yCoordinate']], 'k--')

结果将是

如果您想要更像熊猫的东西，请尝试以下操作：

import seaborn as sns
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
# Replace the following with your own dataframe
# Events=pd.read_csv('StartFinish.csv')
np.random.seed(0)
Events = np.random.randint(0, 10, size=(24, 2))
Events = pd.DataFrame(Events, columns=['xCoordinate', 'yCoordinate'])
Events['Event Type'] = 'Start' 
Events.loc[1::2, 'Event Type'] = 'Finish'

Scatter = sns.scatterplot(x='xCoordinate', y='yCoordinate', hue='Event Type', data=Events)
Scatter

Events['group'] = Events.index//2
Events.groupby('group').apply(lambda x: plt.plot([x.loc[0, 'xCoordinate'], x.loc[1, 'xCoordinate']], 
                                                 [x.loc[0, 'yCoordinate'], x.loc[1, 'yCoordinate']], 'k--'));