Pandas - 创建数据框答案

【问题标题】：Pandas - Create dataframePandas - 创建数据框
【发布时间】：2021-12-02 15:16:18
【问题描述】：

我正在尝试从 csv 文件创建数据框，有多个列和行。其中一列有“是”或“否”。我只希望数据框包含“是”的行。有人可以告诉我如何编写这段代码吗？提前致谢。

【问题讨论】：

这能回答你的问题吗？ conditional row read of csv in pandas
你也可以试试 df.loc[df['column_name'] == 'yes']
Python 和 Pandas 有多个选项可用于在特定条件下进行过滤。很快，这是我刚刚找到你的一个页面towardsdatascience.com/…，但是你可以在 google 上找到很多可以教你其他很酷的方法的页面。不要限制自己学习 Pandas 的所有酷东西。

【解决方案1】：

您可以读取文件然后过滤数据框以仅获取“是”行。例如：

df = pd.read_csv("data.csv")
df = df[df.column == 'yes']

【讨论】：

【解决方案2】：

这里有一些可以帮助您的方法。假设你的列名是choice，你的数据框名是df

df_new = df[df['choice'] == 'yes']

在这种情况下，如果您运行 df_new，您将获得只有 yes 的数据报。

与下面的代码相同。

mask = df['choice'] == 'yes'
  
# new dataframe with selected rows
df_new = pd.DataFrame(df[mask])

你也可以试试这个：

# condition with df.values property
mask = df['choice'].values == 'yes'
  
# new dataframe
df_new = df[mask]
  
print(df_new)

【讨论】：