合並單個 pandas 數據框中的兩列

Question

我有以下數據框：

df = pd.DataFrame({'id': [0.1, 0.2, 0.3, 0.4],'A': [1,2, np.NaN, np.NaN], 'A1': [np.NaN, np.NaN, 3,4]})

我希望將 A1 合並到 A（刪除 A1）中，結果應該是：

df = pd.DataFrame({'id': [0.1, 0.2, 0.3, 0.4, 0.1, 0.2, 0.3, 0.4],'A': [1,2, np.NaN, np.NaN, np.NaN, np.NaN, 3,4]})

感謝任何幫助。

Answer 1

您可以concat兩列：

pd.concat([df[['id', 'A']], df[['id', 'A1']].rename(columns={'A1': 'A'})],
          ignore_index=True)

或者melt rename / drop ：

df.melt('id').rename({'value': 'A'}).drop(columns='variable')

output：

    id    A
0  0.1  1.0
1  0.2  2.0
2  0.3  NaN
3  0.4  NaN
4  0.1  NaN
5  0.2  NaN
6  0.3  3.0
7  0.4  4.0

Answer 2

一種選擇是使用pyjanitor的pivot_longer ； 列有一個模式，它們都以A開頭，我們可以在重塑中利用它來發揮我們的優勢：

# pip install pyjanitor
import janitor
import pandas as pd
df.pivot_longer('id', names_to = 'A', names_pattern=['A'])

    id    A
0  0.1  1.0
1  0.2  2.0
2  0.3  NaN
3  0.4  NaN
4  0.1  NaN
5  0.2  NaN
6  0.3  3.0
7  0.4  4.0