python - 如何在数据框中存储条件输出

Question

我正在尝试将 if 条件的输出存储到 Dataframe。下面给出的是我正在尝试的:

import os

filename = "Desktop/sales/10-05-2018"
#check file exists
if(os.path.exists(filename)):
  print("Files received")
else:
  print("No files received")

我不想打印输出，而是想将输出存储到 Dataframe 中。任何人都可以对此提出建议。谢谢。

Answer 1

这是一种将此类映射存储在数据框中的方法。

import os, pandas as pd

df = pd.DataFrame(columns=['filename', 'exists'])

df['file'] = ['file1.csv', 'file2.csv', 'file3.csv']
df['exists'] = df['file'].map(os.path.exists)

这将在一列中创建一个文件名数据框，在另一列中创建一个 bool 系列，指示文件是否存在。

如果文件名是从可迭代对象中检索的，您应该在构建数据框之前先聚合到列表列表中。在这种情况下，不断附加到现有数据框是低效的。

lst = ( ... some iterable ... )
lst_of_lst = [[f, os.path.exists(f)] for f in lst]

df = pd.DataFrame(lst_of_lst, columns=['filename', 'exists'])