如何选择在Kedro CSVLocalDataSet中加载哪些列?

我有一个csv文件,看起来像

a,b,c,d
1,2,3,4
5,6,7,8

,我想将其作为Kedro CSVLocalDataSet加载,但是我不想读取整个文件。我只需要几列(例如ab)。

我可以指定要读取/加载的列列表吗?

quanta1 回答:如何选择在Kedro CSVLocalDataSet中加载哪些列?

CSVLocalDataSet使用pandas.read_csv,该参数带有“ usecols”参数。可以使用load_args数据集参数轻松代理(所有数据集都支持通过load_argssave_args传递的其他参数):

my_cool_data:
  type: CSVLocalDataSet
  filepath: data/path.csv
  load_args: 
    usecols: ['a','b']

还请注意,相同的参数适用于任何基于熊猫的数据集。

本文链接:https://www.f2er.com/3137901.html

大家都在问