在groupby和loc之后存储值

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

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

  1. import pandas as pd
  2. import numpy as np
  3. df = pd.DataFrame(dict(Gender=['M', 'F', 'M', 'F',
  4. 'F', 'M', 'M', np.nan],
  5. State=['NY','IL', 'NY', 'CA',
  6. 'IL', 'CA', 'CA', 'IL'],
  7. col_1=[10,20,30,np.nan,40,50,60,70],
  8. col_2=[1,6,2,4,6,9,np.nan,3]))
  9. d = df.groupby(['Gender','State']).sum()
  10. d.loc['F']

执行结果

  1. col_1 col_2
  2. State
  3. CA 0.0 4.0
  4. IL 60.0 12.0

预期结果( Dataframe 格式)

  1. a =
  2. col_1 col_2
  3. 0.0 4.0
  4. 60.0 12.0
fxnxkyjh

fxnxkyjh1#

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

  1. print(d.loc['F'].reset_index(drop=True))
  2. col_1 col_2
  3. 0 0.0 4.0
  4. 1 60.0 12.0

相关问题