矢量化方法,用于创建并向csv写入任意数量的数据帧

2024-10-06 06:58:49 发布

您现在位置:Python中文网/ 问答频道 /正文

目前,我有如下数据帧:

          id  temp1  temp2
9       10.0   True  False
10      10.0   True  False
11      10.0  False   True
12      10.0  False   True
17      15.0   True  False
18      15.0   True  False
19      15.0   True  False
20      15.0   True  False
21      15.0  False  False
33      27.0   True  False
34      27.0   True  False
35      27.0  False   True
36      27.0  False  False
40      31.0   True  False
41      31.0  False   True 
           .
           .
           .

我用以下命令对表格进行排序:

u = coinc.groupby('id')
m = u.temp1.any() & u.temp2.any()
res = df.loc[coinc.id.isin(m[m].index), ['id']]

检查是否在具有相同id的任何组中,temp1列和temp2列中是否都存在true。如果有,它将进入新的“res”数据帧

现在,我有一个数据帧,看起来像这样:

         id  temp1_0  temp2_0  temp1_1  temp2_1
9      10.0    False    False     True    False
10     10.0    False    False     True    False
11     10.0    False     True    False    False
12     10.0    False     True    False    False
17     15.0     True    False    False    False
18     15.0     True    False    False    False
19     15.0    False    False     True    False
20     15.0    False    False     True    False
21     15.0    False    False    False    False
33     27.0    False    False     True    False
34     27.0    False    False     True    False
35     27.0    False     True    False    False
36     27.0    False    False    False    False
40     31.0    False    False     True    False
41     31.0    False     True    False    False

除此之外,在现实中,我将有任意数量的临时列,但总是以两个一组的方式,如上面所述

我想知道是否有一种矢量化的方法可以实现上述相同的操作,但对于每组两个临时列(即,对于上面的示例,它将生成两个单独的数据帧),然后根据它们的名称将它们输出到csv(…_0.csv,…_1.csv,等等)


Tags: csv数据命令idfalsetruedf排序