UDF用于查找表中的元素

Question

我需要创建一个用户定义函数，该函数将接受OwnerID作为输入，并输出带有相关PetID的向量。

    import matplotlib.pyplot as plt
    import numpy as np
    import pandas as pd
    pets = pd.read_csv('pets (1).csv')
    def OwnerFunc (OwnerID):
        print ("OwnerID: ", OwnerID)
        return;
    OwnerFunc(OwnerID=pets["PetID"])

pets df的列包含PetID，名称，种类，年龄和OwnerID。

我将它放到将返回所有PetID的位置，但没有任何特定的关联。我感到很抱歉，我对python还是很陌生。

Answer 1

保留函数名称的小写形式，例如def foo()。

pets = pd.read_csv('pets (1).csv')
def owner_func (OwnerID):
    print ("OwnerID: ", OwnerID)
    return pets["PetID"][pets["OwnerID"]==OwnerID]

我假设您要返回与所有者名称关联的宠物名称。如果不是这种情况，请告诉我。