从数据框中返回列值的 Python 函数

Question

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

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']]

希望这有帮助！