在 dask 中将 csv 导入数据框时命名列

Question

I would like to name columns when I import a csv to a dataframe with dask in Python.The code I use looks like this:

 for i in range(1, files + 1): filename = str(i) + 'GlobalActorsHeatMap.csv' runs[i] = dd.read_csv(filename, header=None)

I would like to use an array with names for each column:

names = ['tribute', 'percent_countries_active', 'num_wars', 'num_tributes', 'war', 'war_to_tribute_ratio', 'US_wealth', 'UK_wealth', 'NZ_wealth' ]

Is this possible to do directly?

Answer 1

Just use the names argument for the read_csv

names = [...]
dd.read_csv(filename, header=None, names=names)

Read more here