gpt4 book ai didi

python - 在 Pandas Read_CSV 中使用 UseCols 时按指定顺序保留列

转载 作者:太空狗 更新时间:2023-10-29 17:19:09 26 4
gpt4 key购买 nike

我有一个包含 50 列数据的 csv 文件。我正在使用 Pandas read_csv 函数提取这些列的子集,使用 usecols 参数来选择我想要的列:

cols_to_use = [0,1,5,16,8]
df_ret = pd.read_csv(filepath, index_col=False, usecols=cols_to_use)

问题是 df_ret 包含正确的列,但不是按照我指定的顺序。它们按升序排列,因此 [0,1,5,8,16]。 (顺便说一句,列号可以在每次运行时改变,这只是一个例子。)这是一个问题,因为代码的其余部分具有“正确”顺序的数组,我宁愿不必重新排序所有他们中的。

是否有任何聪明的 pandas 方法可以按照指定的顺序拉入列?任何帮助将不胜感激!

最佳答案

您可以重复使用相同的 cols_to_use 列表来按所需顺序选择列:

df_ret = pd.read_csv(filepath, index_col=False, usecols=cols_to_use)[cols_to_use]

关于python - 在 Pandas Read_CSV 中使用 UseCols 时按指定顺序保留列,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/40024406/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com