Блог

Например, это фрейм данных, с которого я начинаю, называется df1

        shoe       brand         price
0     Jordan       1             50
1     Kyrie        1             75
2     Harden       2             50
3     Kobe         1             50
4     Curry        3             75
 

Теперь я хочу создать функцию, которая использует фрейм данных, столбец бренда и столбец цены для возврата нового фрейма данных. Это то, с чем я работаю до сих пор:

 def dfFunction(dataframe, shoeBrand, priceShoe):
     df2 = dataframe["shoeBrand","priceShoe"]
 

И вот как это будет работать:

 print(dfFunction(df1, 1, 50)

     shoe     brand     price
0   Jordan      1         50
1   Kobe        1         50
 

Поэтому в основном я хочу создать новый фрейм данных, который учитывает бренд и цену при создании нового фрейма данных с использованием pandas.

Другая версия, использующая .query :

 def dfFunction(df, shoeBrand, priceShoe):
    return df.query(f"brand == {shoeBrand} amp; price == {priceShoe}")


print(dfFunction(df1, 1, 50))
 

С принтами:

      shoe  brand  price
0  Jordan      1     50
3    Kobe      1     50