Question

我想删除列ID，并创建两个数据帧，其中一个是唯一行，另一个是重复的行。我的代码在下面，我想要的是将列ID添加到每个dataframe（join）。

d = {'id': ["i1", "i2", "i3", "i4", "i5"], 'x1': [13, 13, 61, 61, 61], 'x2': [10, 10, 13, 13, 13], 'x3': [12, 12, 2, 22, 2], 'x4': [24, 24, 9, 12, 9]}
df = pd.DataFrame(data=d)
del df['id']
dfduplicated = df[df.duplicated()]
dfUNIC= df.drop_duplicates(keep='first')

Answer 1

用id删除drop，用DataFrame.duplicated测试重复项，用boolean indexing过滤原始数据：

m = df.drop('id', axis=1).duplicated()
dfduplicated = df[m]
print (dfduplicated)
   id  x1  x2  x3  x4
1  i2  13  10  12  24
4  i5  61  13   2   9

然后将反掩码用作~：

dfUNIC= df[~m]
print (dfUNIC)
   id  x1  x2  x3  x4
0  i1  13  10  12  24
2  i3  61  13   2   9
3  i4  61  13  22  12

Answer 2

我会做cumcount

s=df.groupby(list(set(df)-{'id'})).cumcount()
df1=df[s==0].copy()
df2=df.drop(df1.index)
df1
Out[113]: 
   id  x1  x2  x3  x4
0  i1  13  10  12  24
2  i3  61  13   2   9
3  i4  61  13  22  12
df2
Out[114]: 
   id  x1  x2  x3  x4
1  i2  13  10  12  24
4  i5  61  13   2   9

数据框行的访问索引

2 个答案: