如何将数据帧转换为从-到对?

问题描述 投票:0回答:2

如果我有数据框:

>>> import pandas as pd
>>> df = pd.DataFrame([
...     ['A', 'B', 'C', 'D'],
...     ['E', 'B', 'C']
... ])
>>> df
   0  1  2     3
0  A  B  C     D
1  E  B  C  None
>>>  

我应该将数据框转换为两列格式:

x, y
-----
A, B
B, C
C, D
E, B
B, C

对于每一行,从左到右,取两个邻居值并成对。如果您将每一行都视为一条路径,这是一种从头到尾的方法。

如何进行转换?

python pandas transformation shift
2个回答
2
投票

我们可以用explodezip

s=pd.DataFrame(df.apply(lambda x : list(zip(x.dropna()[:-1],x.dropna()[1:])),axis=1).explode().tolist())
Out[336]: 
   0  1
0  A  B
1  B  C
2  C  D
3  E  B
4  B  C

0
投票

准备数据也有帮助:

import pandas as pd

inp = [['A', 'B', 'C', 'D'],
       ['E', 'B', 'C']]

# Convert beforehand
inp2 = [[[i[k], i[k+1]] for k in range(len(i)-1)] for i in inp]
inp2 = inp2[0] + inp2[1]

df = pd.DataFrame(inp2)
print(df)

输出:

   0  1
0  A  B
1  B  C
2  C  D
3  E  B
4  B  C
© www.soinside.com 2019 - 2024. All rights reserved.