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

Question

当我在 Python 中使用 dask 将 csv 导入到数据框时，我想命名列。我使用的代码如下所示：

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

我想为每列使用一个带有名称的数组：

名称 = ['tribute'、'percent_countries_active'、'num_wars'、'num_tributes'、'war'、'war_to_tribute_ratio'、'US_wealth'、'UK_wealth'、'NZ_wealth']

这可以直接做吗？

Answer 1

只需使用names参数read_csv

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

在这里阅读更多