Python按列名称获取熊猫数据

时间:2019-09-28 04:07:26

标签: python pandas csv

我正在尝试通过名为“模型”的列提取每个熊猫的数据。 下面的代码可以正常工作,但是如果我想使用x = R1200GS(不带”)如何编写代码?

由于我需要从其他函数传递此var,因此请告诉我是否有一种好的方法。 谢谢。

import pandas as pd

def get_database(z):
    df=pd.read_csv("database.csv")
    df=df.query("model==@z")
    return df

x = 'R1200GS'

get_database(x)

1 个答案:

答案 0 :(得分:0)

目前无法发表评论,因此将在此处发布。不确定这是否会奏效。除非有一个名为x = R1200GS的对象,否则R1200GS会引发错误。

很好奇:使用字符串有什么问题?您能详细说明一下吗?