时间戳

首页 > 解决方案 > 如何存储熊猫列名称以进行编程访问？

python - 如何存储熊猫列名称以进行编程访问？

问题描述

我想知道存储熊猫列名的最佳做法是什么。我将数据加载到数据框中，然后按名称访问列。

但是，我不想将字符串硬编码到我的程序中，因为没有 linting 支持，如果我更改列名，我需要手动寻找字符串。

目前，我想出的最好的方法是使用枚举。

class Col(enum.Enum):
    col1 = 1
    col2 = 2
    col3 = 3

然后访问列df[Col.col1.name]

优点： - 定义良好的对象Col，而不是单独的常量

缺点： - 冗长且无信息的部分（Col.和.name）

有解决这个问题的标准方法吗？有没有更好的办法？

类具有熊猫也支持的点表示法。但是，缺点是没有对此的 linting 支持。是否可以使自动完成（在 PyCharm 中）工作df.col1？

标签： pythonpandas

解决方案

推荐阅读