如何使用 python 和 pandas 將多列合並為具有多條記錄的一列？

Question

我有一個數據需要轉換才能獲得 2 列，共 4 列：

data = [['123', 'Billy', 'Bill', 'Bi'],
        ['234', 'James', 'J', 'Ji'],
        ['543', 'Floyd', 'Flo', 'F'],
]

processed_data = ?

needed_df = pandas.DataFrame(processed_data, columns=['Number', 'Name'])

我期望以下行為：

['123', 'Billy']
['123', 'Bill']
['123', 'Bi']
['234', 'James']
['234', 'J']
['234', 'Ji']

我試圖在 for 循環中使用 for 但得到了錯誤的結果：

for row in df.iterrows():
    for col in df.columns:
        new_row = ...
        processed_df = pandas.concat(df, new_row)

這樣的結構給出了太大的結果

使用 sql 的類似問題：

如何將SQL中的多條記錄拆分為一列？

Answer 1

讓我們使用列表理解來創建一對 Name 和 Number 然后創建一個新的 dataframe

pd.DataFrame([[x, z] for x, *y in data for z in y], columns=['Number', 'Name'])

---

  Number   Name
0    123  Billy
1    123   Bill
2    123     Bi
3    234  James
4    234      J
5    234     Ji
6    543  Floyd
7    543    Flo
8    543      F

Answer 2

或者，您可以將現有數據轉換為 dataframe 然后執行 pandas dataframe 使用melt重塑：

import pandas as pd

data = [['123', 'Billy', 'Bill', 'Bi'],
        ['234', 'James', 'J', 'Ji'],
        ['543', 'Floyd', 'Flo', 'F'],
]


df = pd.DataFrame(data)

df.melt(0).sort_values(0)

Output：

     0  variable  value
0  123         1  Billy
3  123         2   Bill
6  123         3     Bi
1  234         1  James
4  234         2      J
7  234         3     Ji
2  543         1  Floyd
5  543         2    Flo
8  543         3      F