7

假设我有一个my_df列重复的数据框,例如

foo bar foo hello
0   1   1   5
1   1   2   5
2   1   3   5

我想创建另一个平均重复的数据框:

foo bar hello
0.5   1   5
1.5   1   5
2.5   1   5

我怎样才能在熊猫中做到这一点?

到目前为止,我已经设法识别出重复项:

my_columns = my_df.columns
my_duplicates = print [x for x, y in collections.Counter(my_columns).items() if y > 1]

我不知道如何让 Pandas 平均它们。

4

1 回答 1

5

您可以groupby列索引并采用mean

In [11]: df.groupby(level=0, axis=1).mean()
Out[11]:
   bar  foo  hello
0    1  0.5      5
1    1  1.5      5
2    1  2.5      5

一个有点棘手的例子是如果有一个非数字列:

In [21]: df
Out[21]:
   foo  bar  foo hello
0    0    1    1     a
1    1    1    2     a
2    2    1    3     a

以上将引发:DataError: No numeric types to aggregate绝对不会因为效率而赢得任何奖励,但在这种情况下,这里有一个通用的方法:

In [22]: dupes = df.columns.get_duplicates()

In [23]: dupes
Out[23]: ['foo']

In [24]: pd.DataFrame({d: df[d] for d in df.columns if d not in dupes})
Out[24]:
   bar hello
0    1     a
1    1     a
2    1     a

In [25]: pd.concat(df.xs(d, axis=1) for d in dupes).groupby(level=0, axis=1).mean()
Out[25]:
   foo
0  0.5
1  1.5
2  2.5

In [26]: pd.concat([Out[24], Out[25]], axis=1)
Out[26]:
   foo  bar hello
0  0.5    1     a
1  1.5    1     a
2  2.5    1     a

我认为要带走的是避免列重复......或者也许我不知道我在做什么。

于 2013-05-21T20:17:08.020 回答