连接pandas DataFrame中的行

jpfvwuh4 于 2023-06-20 发布在其他

关注(0)|答案(1)|浏览(72)

我有一个DataFrame定义为：

test_df = pd.DataFrame(
    {"col_one": [1, 2, 3, 4, 5], "col_two": ["one", "two", "three", "four", "five"]}
).astype(str)

我使用这段代码将所有行变成一行，并将值串联起来：

for c in test_df.columns:
    test_df[c] = ",".join(test_df[c].values)
print(test_df)

打印语句的结果如下：

但我想要的结果是这样的：

我怎么才能做到呢？

1条答案

我会使用 selfcat：

out = test_df.apply(lambda x: x.str.cat(sep=",")).to_frame().T

或者使用agg的这个变体：

out = test_df.agg(",".join).to_frame().T

输出：

print(out)

     col_one                  col_two
0  1,2,3,4,5  one,two,three,four,five