我有一只两栏的熊猫
source | target
----------------
1 3
2 4
2 1这表示图的边缘列表。就目前情况而言,图是有向的。我想把它转换成一个无向图,所以我需要“互补”边,以便获得:
source | target
----------------
1 3
3 1
2 4
4 2
2 1
1 2我是否可以在Pandas中直接做到这一点,而无需使用图形库(如networkx )?
发布于 2021-01-10 17:23:13
我是否可以在Pandas中直接这样做,而无需使用图形库(如networkx )。
我们可以使用df.iloc反转df,然后使用df.append追加
out = df.append(pd.DataFrame(df.iloc[:,::-1].to_numpy(),index=df.index
,columns=df.columns)).sort_index().reset_index(drop=True)或者使用np.flip
(df.append(pd.DataFrame(np.flip(df.to_numpy(),axis=1),
index=df.index,columns=df.columns)).sort_index().reset_index(drop=True))print(out)
source target
0 1 3
1 3 1
2 2 4
3 4 2
4 2 1
5 1 2注意:如果有更多的列,您可以将dataframe的一个子集作为sub_df = df[['source','target']],并将df替换为代码的sub_df i。
@ShubhamSharma提出的另一个使用set_axis重命名列的方法是:
(df.append(df.iloc[:, ::-1].set_axis(df.columns, 1,inplace=False))
.sort_index().reset_index(drop=True))https://stackoverflow.com/questions/65656138
复制相似问题