python - Pandas 数据框中每两列的总和

Question

当我使用 Pandas 时，我遇到了一个问题。我的任务是这样的:

df=pd.DataFrame([(1,2,3,4,5,6),(1,2,3,4,5,6),(1,2,3,4,5,6)],columns=['a','b','c','d','e','f'])
Out:
    a b c d e f
0   1 2 3 4 5 6
1   1 2 3 4 5 6 
2   1 2 3 4 5 6

我想要做的是输出数据框如下所示:

Out:
    s1   s2   s3
0   3    7    11
1   3    7    11
2   3    7    11

即分别对(a,b),(c,d),(e,f)列求和，将结果列名重命名为(s1,s2,s3)。谁能帮助解决 Pandas 中的这个问题？非常感谢。

Answer 1

1) 通过提供 axis=1 执行 groupby w.r.t 列。根据@Boud 的评论，您只需对分组数组进行一些小的调整就可以得到您想要的:

df.groupby((np.arange(len(df.columns)) // 2) + 1, axis=1).sum().add_prefix('s')

根据以下条件进行分组:

np.arange(len(df.columns)) // 2
# array([0, 0, 1, 1, 2, 2], dtype=int32)

---

2) 使用 np.add.reduceat这是一个更快的选择:

df = pd.DataFrame(np.add.reduceat(df.values, np.arange(len(df.columns))[::2], axis=1))
df.columns = df.columns + 1
df.add_prefix('s')

时序约束:

对于 100 万行跨越 20 列的 DF:

from string import ascii_lowercase
np.random.seed(42)
df = pd.DataFrame(np.random.randint(0, 10, (10**6,20)), columns=list(ascii_lowercase[:20]))
df.shape
(1000000, 20)

def with_groupby(df):
    return df.groupby((np.arange(len(df.columns)) // 2) + 1, axis=1).sum().add_prefix('s')

def with_reduceat(df):
    df = pd.DataFrame(np.add.reduceat(df.values, np.arange(len(df.columns))[::2], axis=1))
    df.columns = df.columns + 1
    return df.add_prefix('s')

# test whether they give the same o/p
with_groupby(df).equals(with_groupby(df))
True

%timeit with_groupby(df.copy())
1 loop, best of 3: 1.11 s per loop

%timeit with_reduceat(df.copy())   # <--- (>3X faster)
1 loop, best of 3: 345 ms per loop