Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import pandas as pd
- import random
- data_1 = random.sample(range(200, 500), 20)
- data_2 = random.sample(range(100, 500), 20)
- data_3 = random.sample(range(200, 500), 20)
- df_1 = pd.DataFrame(data_1, columns=["num_1"])
- df_2 = pd.DataFrame(data_2, columns=["num_2"])
- df_3 = pd.DataFrame(data_3, columns=["num_3"])
- df_1 = df_1.join([df_2, df_3], how='left')
- df_1['max'] = '-'
- list_columns = df_1.columns
- # print(list_columns)
- display(df_1)
- for i in range(len(df_1)):
- list_values = list(df_1.loc[i, "num_1":"num_3"])
- max_value = max(list_values)
- index_max = list_values.index(max_value)
- # print(max_value, index_max)
- df_1.loc[i, "max"] = list_columns[index_max]
- display(df_1)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement