python - 如何将数据帧转换为所需的格式？

Question

我正在寻找将数据帧转换为特定格式。

示例数据框如下:

Col1
a
b
c

我想通过分成两列将上述数据框转换为以下格式:

Col1  Col2
a      a
a      b
a      c
b      b
b      c
c      c

我正在尝试获取 Col1 列的所有组合。

Answer 1

你可以尝试itertools.combinations_with_replacement :

from itertools import combinations_with_replacement as comb
df = pd.DataFrame(list(comb(df['Col1'], 2)), columns=['Col1', 'Col2'])
print(df)

输出:

  Col1 Col2
0    a    a
1    a    b
2    a    c
3    b    b
4    b    c
5    c    c
>>> 

编辑:

感谢@QuangHoang的评论，他提到对于更高版本(Quang Hoang有1.1.4)，您可以这样做:

df = pd.DataFrame(comb(df['Col1'], 2), columns=['Col1', 'Col2'])

没有list(...)。而对于较低版本，您会得到:

TypeError: data argument can't be an iterator