在groupby和loc之后存储值

ubby3x7f  于 2021-09-13  发布在  Java
关注(0)|答案(1)|浏览(427)

使用 df 选择两列 col_1col_2 之后 groupbyloc 并存储在变量 a Dataframe 格式
python代码

import pandas as pd
import numpy as np

df = pd.DataFrame(dict(Gender=['M', 'F', 'M', 'F',
                           'F', 'M', 'M', np.nan],
                   State=['NY','IL', 'NY', 'CA',
                          'IL', 'CA', 'CA', 'IL'],
                   col_1=[10,20,30,np.nan,40,50,60,70],
                   col_2=[1,6,2,4,6,9,np.nan,3]))

d = df.groupby(['Gender','State']).sum()
d.loc['F']

执行结果

col_1    col_2
State       
CA     0.0      4.0
IL     60.0     12.0

预期结果( Dataframe 格式)

a = 
           col_1    col_2

           0.0      4.0
           60.0     12.0
fxnxkyjh

fxnxkyjh1#

您可以在获取grupby数据后重置索引 d .

print(d.loc['F'].reset_index(drop=True))

   col_1  col_2
0    0.0    4.0
1   60.0   12.0

相关问题