我的问题类似于Pandas: remove reverse duplicates from dataframe但我有一个额外的要求。我需要维护行值对。
例如:
我有datawhere columnA对应 columnC和 columnB对应 column D。
import pandas as pd
# Initial data frame
data = pd.DataFrame({'A': [0, 10, 11, 21, 22, 35, 5, 50],
'B': [50, 22, 35, 5, 10, 11, 21, 0],
'C': ["a", "b", "r", "x", "c", "w", "z", "y"],
'D': ["y", "c", "w", "z", "b", "r", "x", "a"]})
data
# A B C D
#0 0 50 a y
#1 10 22 b c
#2 11 35 r w
#3 21 5 x z
#4 22 10 c b
#5 35 11 w r
#6 5 21 z x
#7 50 0 y a
Run Code Online (Sandbox Code Playgroud)
我想删除存在于列重复A和B但我需要保存在列其对应的字母值C和D。
我在这里有一个解决方案,但有没有更优雅的方法来做到这一点?
# Desired data frame
new_data = pd.DataFrame()
# Concat numbers and corresponding letters
new_data['AC'] = data['A'].astype(str) + ',' + data['C']
new_data['BD'] = data['B'].astype(str) + ',' + data['D']
# Drop duplicates despite order
new_data = new_data.apply(lambda r: sorted(r), axis = 1).drop_duplicates()
# Recreate dataframe
new_data = pd.DataFrame.from_items(zip(new_data.index, new_data.values)).T
new_data = pd.concat([new_data.iloc[:,0].str.split(',', expand=True),
new_data.iloc[:,1].str.split(',', expand=True)], axis=1)
new_data.columns=['A', 'B', 'C', 'D']
new_data
# A B C D
#0 0 a 50 y
#1 10 b 22 c
#2 11 r 35 w
#3 21 x 5 z
Run Code Online (Sandbox Code Playgroud)
编辑技术输出应该是这样的:
new_data.columns=['A', 'C', 'B', 'D']
new_data
# A B C D
#0 0 a 50 y
#1 10 b 22 c
#2 11 r 35 w
#3 21 x 5 z
Run Code Online (Sandbox Code Playgroud)
我认为你可以用stack,drop_duplicates和unstack:
data.set_index(['A','B']).stack().drop_duplicates().unstack().reset_index()
A B C D
0 0 50 a y
1 10 22 b c
2 11 35 r w
3 21 5 x z
Run Code Online (Sandbox Code Playgroud)
| 归档时间: |
|
| 查看次数: |
1050 次 |
| 最近记录: |