使用 'loc' 选择多列

Question

我有一个包含 3 列的新闻数据集：lead_paragraph, _id, web_url。以下代码返回与 id 和 News 匹配的行lead_paragraph。ds = pd.read_csv("nytimes.csv")def item(id):    return ds.loc[ds['_id'] == id]['lead_paragraph'].tolist()[0]如何获得web_url也在lead_paragraph列表中？尝试一种解决方案后出现错误return ds.loc[ds['_id'] == id], ['web_url', 'lead_paragraph']].tolist()                                                             ^SyntaxError: invalid syntax

Answer 1

使用df.loc[ row indexer , columns indexer ]，

df.loc[df['_id'] == id, ['web_url', 'lead_paragraph']].values[0].tolist()

其中行索引器是一个布尔系列，列索引器是一个列标签列表。