vmamontov

concatenate_data_few_columns

Nov 12th, 2021 (edited)
1,335
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.10 KB | None | 0 0
  1. import pandas as pd
  2. import numpy as np
  3.  
  4. df = pd.DataFrame(np.arange(25).reshape(-1, 5),
  5.                   columns=['a', 'b', 'pref_c', 'pref_d', 'pref_e'])
  6.  
  7. columns = list(df.filter(like='pref_', axis=1).columns)
  8.  
  9. df['merge_str'] = ""
  10. df['merge_int'] = 0
  11.  
  12. for column in columns:
  13.     df['merge_str'] += df[column].astype(str)
  14.     df['merge_int'] += df[column]
  15.  
  16. print(df)
  17.  
  18. # Задача:
  19. # объединить несколько столбцов датафрейма с определенным префиксом
  20.  
  21. #     a   b  pref_c  pref_d  pref_e
  22. # 0   0   1       2       3       4
  23. # 1   5   6       7       8       9
  24. # 2  10  11      12      13      14
  25. # 3  15  16      17      18      19
  26. # 4  20  21      22      23      24
  27.  
  28. # Решение:
  29.  
  30. #     a   b  pref_c  pref_d  pref_e merge_str  merge_int
  31. # 0   0   1       2       3       4       234          9
  32. # 1   5   6       7       8       9       789         24
  33. # 2  10  11      12      13      14    121314         39
  34. # 3  15  16      17      18      19    171819         54
  35. # 4  20  21      22      23      24    222324         69
Add Comment
Please, Sign In to add comment