Я запускаю цикл для таблицы slice на основе значения данного словаря. Но я хотел бы создать одну таблицу для каждого цикла и сохранить таблицу, используя ключ словаря. В следующем примере я хотел бы поместить "ключ" в имя таблицы df_slice, например df_slice_loc1, df_slice_loc2
import numpy as np
import pandas as pd
df = pd.DataFrame({'group': ['A', 'B', 'C', 'D', 'E']
,'scoreA': np.random.randn(5)
,'scoreB': np.random.randn(5)})
loc_d = {"loc1":2, "loc2":3}
for key, value in loc_d.items():
df_slice = df.iloc[:value,]
Например, используя понимание словаря:
df_slice = {key: df.iloc[:value] for key, value in loc_d.items()}
print(df_slice)
{'loc1': group scoreA scoreB
0 A 0.139625 -0.321823
1 B 2.137781 0.103702,
'loc2': group scoreA scoreB
0 A 0.139625 -0.321823
1 B 2.137781 0.103702
2 C -1.000334 0.680955}
Затем войдите в срез через, например, df_slice['loc2']
.
loc_d
и сопоставить их с соответствующими имdf.iloc[value,]
s?