Python函数从数据框

Question

我想创建一个带有两个参数的函数，这些参数从Names列返回一些特定的名称。第一个参数是现有数据帧，第二个参数是name。我知道如何从行中提取特定值但是将它放在多参数函数中是我的问题：

def returnDataForOneName(namesDF, name):
    for string in name:
    ?--> return [string.values() for string in namesDF]

我想用我现有的一个csv文件测试该函数：

  import pandas as pd
    datapath = 'C:\Users\namefile.csv'
    namesDF = pd.read_csv(datapath) 
    newDF = returnDataForOneName(nameDF, 'Tim')

Answer 1

您可以使用简单的索引解决此问题，无需函数：

newDf = nameDf[nameDf['name'] == 'Tim']]

希望这有帮助！