Pandas read_csv()有条件地跳过标题行

2 回答

• 3

  如果 CSV 文件中的标题遵循类似的模式，您可以执行一些简单的操作，例如在确定是否跳过第一行之前嗅出第一行。

  filename = '/path/to/file.csv'
  skiprows = int('Created in' in next(open(filename)))
  df = pd.read_csv(filename, skiprows=skiprows)
  

  ---

  好的做法是使用上下文管理器，所以你也可以这样做：

  filename = '/path/to/file.csv'
  skiprows = 0
  with open(filename, 'r+') as f:
      for line in f:
          if line.startswith('Created '):
              skiprows = 1
          break
  df = pd.read_csv(filename, skiprows=skiprows)
  

  回复于 2024-04-20T21:31:01+08:00
• 0

  在 pandas read_csv 命令中使用'comment'参数时，可以跳过以特定字符开头的行。在您的情况下，您可以使用以下代码跳过以“C”开头的行：

  filename = '/path/to/file.csv'
  pd.read_csv(filename, comment = "C")
  

  回复于 2024-04-20T21:31:01+08:00