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