Pandas DataFrame 的条件平均值

Question

我有一个包含多列房地产销售数据的数据框。我想按邮政编码查找所有 1bed-1bath 销售的每平方英尺平均价格 'ppsf'。这是我的尝试（字典中的每个键都是一个邮政编码）：

bed1_bath1={}
for zip in zip_codes:
    bed1_bath1[zip]= (df.loc[(df['bed']==1) & (df['bath']==1) & (df['zip']==zip)]).mean()

问题在于，这会将数据帧中所有列的平均值添加到字典中。我确信有更好的方法来做到这一点；也许使用 numpy.where？

Answer 1

(df[(df['bed']==1) & (df['bath']==1) & (df['zip']==zip)])['ppsf'].mean() 会这样做。您只需在计算平均值之前选择您感兴趣的列（因此您甚至不需要对其余列进行处理）。