如何使用此代码仅获取我想移动到新数据框的列？ [复制]

Question

我正在尝试从 csv 文件中仅选择 2 列：Body 和 CreatedDate。 CreatedDate 看起来像这样：2018-08-07T12:36:11.000Z。 Body 只是正在完成的工作的文本。一些Body 单元格是空的，所以我只想要其中包含数据的单元格。

我已经尝试使用下面的代码只获取 2 个所需的列：

import pandas as pd
df = pd.read_csv("file.csv")
df1= df['CreatedDate'].map(str) + ' ' + df['Body'].map(str)
print(df1)

我将整个df 打印两次。我看到了：

[10 行 x 15 列] & [15 行 x 10 列]

在每张印刷品的底部。我希望只看到我选择的 2 个列。为什么我会在控制台上看到所有 df 两次？

Answer 1

indexing a dataframe 有很多选项。这个特定的可以在一行上完成。

import pandas as pd
# read the csv into df
df = pd.read_csv("file.csv")
# take only the rows where 'Body' has a value and only columns ['Body', 'CreatedDate']
df = df.loc[df['Body'].notnull(),['Body', 'CreatedDate']]
print(df)

您可能还想阅读pandas.DataFrame.dropna。