使用 '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()其中行索引器是一个布尔系列，列索引器是一个列标签列表。