[英]Extract columns by file name from Pandas dataframe
>>> df
file_name x_c y_c
0 a.tif 7 37
1 a.tif 23 41
2 a.tif 98 21
3 b.tif 74 100
4 b.tif 84 78
5 b.tif 50 10
6 b.tif 1 10
7 c.tif 10 57
8 c.tif 49 15
>>> g = df.groupby("file_name")
>>> variables_names = [f"df_{e}" for e,i in enumerate(g, start=1)]
>>> for name, group in zip(variables_names, g):
globals()[name] = group[1].reset_index(drop=True)
>>> df_1
file_name x_c y_c
0 a.tif 7 37
1 a.tif 23 41
2 a.tif 98 21
>>> df_2
file_name x_c y_c
0 b.tif 74 100
1 b.tif 84 78
2 b.tif 50 10
3 b.tif 1 10
>>> df_3
file_name x_c y_c
0 c.tif 10 57
1 c.tif 49 15
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.