优雅的方式来检查字典中的多个条件

Question

我有一个数据框字典，如下所示

dataFramesDict[sheet_name] = pd.DataFrame({'Date':['2007-05-30','2107-11-30','2207-05-20','2307-05-20'],'Value': [2.4,2.5,2.6,2.7],'Test': ['Height','Weight','SBP','DBP']})

我想做的是创建一个名为Unit的新列，并根据Test列添加值

这是我尝试过的方法，但看起来并不优雅或高效。

if dataFramesDict[sheet_name]['Test'] == 'Height':
    dataFramesDict[sheet_name]['unit'] == 'm'
if dataFramesDict[sheet_name]['Test'] == 'Weight':
    dataFramesDict[sheet_name]['unit'] == 'kg'
if dataFramesDict[sheet_name]['Test'] == 'SBP':
    dataFramesDict[sheet_name]['unit'] == 'millimeter'
if dataFramesDict[sheet_name]['Test'] == 'DBP':
    dataFramesDict[sheet_name]['unit'] == 'millimeter'

请注意， sheet_name是Excel工作表的一部分。 总共我有10多张纸，我想阅读每张纸并检查每张纸中的测试值，并根据“ Test列分配单位值

我希望我的输出如下所示

Answer 1

将map与dict使用

例如：

d = {'Height':"m",'Weight': "kg",'SBP':'mm','DBP':'mm'}
dataFramesDict = pd.DataFrame({'Date':['2007-05-30','2107-11-30','2207-05-20','2307-05-20'],'Value': [2.4,2.5,2.6,2.7],'Test': ['Height','Weight','SBP','DBP']})
dataFramesDict["Unit"] = dataFramesDict["Test"].map(d)
print(dataFramesDict)

输出：

         Date    Test  Value Unit
0  2007-05-30  Height    2.4    m
1  2107-11-30  Weight    2.5   kg
2  2207-05-20     SBP    2.6   mm
3  2307-05-20     DBP    2.7   mm

Answer 2

使用熊猫替换

df['unit'] = df.Test.replace({'Height':"m",'Weight': "kg",'SBP':'mm','DBP':'mm'})

Answer 3

请尝试以下代码。 希望能帮助到你：

unitDict = {'Height':'m', 'Weight': 'Kg', 'SBP':'millimeter', 'DBP':'millimeter'}


def checkUnit(data):
  if unitDict[data] is not None:
    return unitDict[data]

df['Unit']=list(map(checkUnit, df['Test']))

print(df)