Python:使用条件索引

2024-06-01 01:47:30 发布

您现在位置:Python中文网/ 问答频道 /正文

我需要画一条线连接一个数据帧中的两个点(在下表中,一条线连接S01到S02,S01到S04,S03到S04等等),点不能被明确定义,而是通过在数据帧上迭代来定义。每行中的坐标是“到”点的坐标。你知道吗

table

以下是我迄今为止尝试过的:

df = pd.read_csv(1234.csv')

for index, row in df.iterrows():
    x_coords = [df.loc["from", "x"], df.loc["to", "x"]]
    y_coords = [df.loc["from", "y"], df.loc["to", "y"]]
    plt.plot(x_coords, y_coords,'grey', linewidth=0.5)

Tags: csvto数据fromdfforread定义
1条回答
网友
1楼 · 发布于 2024-06-01 01:47:30

贾默找到了答案:

for i in df.index:
   df.set_index('to')
   a = df.at[i,'From'] #row of from value
   b = a - 1 #correct index value of row
   ax = df.at[b, 'x'] #x value of from
   ay = df.at[b, 'y'] #y of from
   plt.plot([ax,df.at[i,'x']],[ay,df.at[i,'y']],'grey', linewidth=0.5)

相关问题 更多 >