Merge nhiều giá trị cột thành một cột trong gấu trúc python

Question

Tôi có một khung dữ liệu gấu trúc như thế này:

Column1 Column2 Column3 Column4 Column5 
0 a  1  2  3  4 
1 a  3  4  5 
2 b  6  7  8 
3 c  7  7   

Những gì tôi muốn làm bây giờ là nhận được một dataframe mới có chứa COLUMN1 và columnA mới. Cột nàyA phải chứa tất cả các giá trị từ cột 2 - (đến) n (trong đó n là số cột từ Cột2 đến cuối hàng) như sau:

Column1 ColumnA 
0 a  1,2,3,4 
1 a  3,4,5 
2 b  6,7,8 
3 c  7,7 

Làm cách nào để tiếp cận tốt nhất vấn đề này? Lời khuyên nào sẽ là hữu ích. Cảm ơn trước!

Answer 1

Bạn có thể gọi apply qua axis=1-apply hàng-khôn ngoan, sau đó chuyển đổi dtype để str và join:

In [153]: 
df['ColumnA'] = df[df.columns[1:]].apply(lambda x: ','.join(x.dropna().astype(int).astype(str)),axis=1) 
df 

Out[153]: 
    Column1 Column2 Column3 Column4 Column5 ColumnA 
0  a  1  2  3  4 1,2,3,4 
1  a  3  4  5  NaN 3,4,5 
2  b  6  7  8  NaN 6,7,8 
3  c  7  7  NaN  NaN  7,7 

Ở đây tôi gọi dropna để thoát khỏi các NaN, tuy nhiên chúng tôi cần phải cast một lần nữa để int vì vậy chúng tôi không kết thúc với phao như str.