如何使用具有相同标签的列从数据框中删除列?

mep*_*ein 3 python dataframe pandas

我有一个数据框,其中一些列标签出现多次(即,一些列具有相同的标签).这引起了我的问题 - 我可能会单独发布更多相关信息,因为有些行为似乎有些奇怪,但在这里我只是想问一下删除其中的一些列.也就是说,对于多次出现的每个列标签,我想删除它前面的第一列以外的所有列.这是一个例子:

In [5]: arr = np.array([[0.0, 1.0, 2.0, 3.0], [4.0, 5.0, 6.0, 7.0]])

In [6]: df = pd.DataFrame(data=arr, columns=['A', 'C', 'E', 'A'])

In [7]: df
Out[7]:
   A  C  E  A
0  0  1  2  3
1  4  5  6  7
Run Code Online (Sandbox Code Playgroud)

如果我使用标签删除列,则删除以该标签为首的所有列:

In [9]: df.drop('A', axis=1)
Out[9]:
   C  E
0  1  2
1  5  6
Run Code Online (Sandbox Code Playgroud)

所以我想我会尝试按列索引删除,但这也删除了该标签所有的所有列:

In [12]: df.drop(df.columns[3], axis=1)
Out[12]:
   C  E
0  1  2
1  5  6
Run Code Online (Sandbox Code Playgroud)

我怎样才能做我想要的,也就是说,对于每个这样的标签,删除除了其中一列之外的所有列?对于上面的例子,我想最终得到:

   A  C  E
0  0  1  2
1  4  5  6
Run Code Online (Sandbox Code Playgroud)

现在我重新标记了列,如下所示:

columns = {}
new_columns = []
duplicate_num = 0
for n in df.columns:
    if n in columns:
        new_columns.append("duplicate%d" % (duplicate_num))
        duplicate_num += 1
    else:
        columns[n] = 1
        new_columns.append(n)
df.columns = new_columns
Run Code Online (Sandbox Code Playgroud)

这对我的需求很好,但它似乎不是最好/最干净的解决方案.谢谢.

编辑:我不知道这是如何与另一个问题重复.首先,它处理重复的列,而不是重复的列标签.另一方面,建议的解决方案涉及转置数据帧(两次),但如前所述,转置大型数据帧效率低下,事实上我正在处理大型数据帧.

Nad*_*ham 5

In [18]:  
df.ix[: , ~df.columns.duplicated()]
Out[18]:
    A   C   E
0   0   1   2
1   4   5   6
Run Code Online (Sandbox Code Playgroud)

说明


In [19]:
~df.columns.duplicated()
Out[19]:
array([ True,  True,  True, False], dtype=bool)
Run Code Online (Sandbox Code Playgroud)

正如你在这里看到的,你首先需要检查列名是否重复,注意我~在函数的开头添加了.然后,您可以使用非重复值对列进行切片