從數據框中返回列值的 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']]

希望這有幫助！