如何打印从 elasticsearch 导出的 CSV 文件中的列名？

Question

以下是从弹性搜索中获取一些数据并将该数据导出到名为“mycsvfile”的 csv 文件的代码。

我想更改列名，以便人类可以阅读。

下面是代码：

from elasticsearch import Elasticsearch
import csv

es = Elasticsearch(["9200"])

# Replace the following Query with your own Elastic Search Query
res = es.search(index="search", body=
                {
                    "_source": ["DTDT", "TRDT", "SPLE", "RPLE"],
                    "query": {
                        "bool": {
                            "should": [
                                {"wildcard": {"CN": "TEST1"}}

                            ]
                        }
                    }
}, size=10)



with open('mycsvfile.csv', 'w') as f:  # Just use 'w' mode in 3.x
    header_present  = False
    for doc in res['hits']['hits']:
        my_dict = doc['_source'] 
        if not header_present:
            w = csv.DictWriter(f, my_dict.keys())
            w.writeheader()
            header_present = True


        w.writerow(my_dict)

当我运行上述查询时，CSV 文件数据如下所示：

DTDT    TRDT    SPLE    SACL    RPLE

20170512    12/05/2017 15:39    1001    0   0

20170512    12/05/2017 15:39    1001    0   0

20170908    08/09/2017 02:42    1001    0   0

20170908    08/09/2017 06:30    1001    0   0

如您所见，列名与查询中的相同，我想在生成文件时为它们提供可读的名称。

有人可以显示并修复我的代码，以便我在 CSV 文件中输入列名吗？

提前谢谢你

Answer 1

如果您不介意或愿意，可以尝试使用 pandas。如果您使用 pandas，您的解决方案是：

import pandas as pd

df = pd.DataFrame.read_csv('mycsvfile.csv')

print(df.columns)

我还想补充一点，如果您的标题情况不同或您的索引列不存在，您可能需要对 df 创建语句进行一些调整。这是文档link：

Answer 2

您想要重命名字典键。 To do that, you can for example pop it to a new column name. 将 DTDT 更改为 date 的示例。

my_dict['date'] = my_dict.pop('DTDT')

我认为您可以弄清楚如何重命名其他列。之后，您可以在 for 循环中调用 w.writerow。

Answer 3

那怎么样？定义一个顶部带有翻译的字典，例如在初始化您的 es 客户端之后，在第 4 或第 5 行左右：

readableColumnNames = {"DTDT" : "Date", "BLI" : "Blub"}

然后替换这行代码：

w = csv.DictWriter(f, [readableColumnNames[colName] for colName in my_dict.keys()])

说实话，没有对此进行测试，但它应该可以完成这项工作，而且简单透明。您可能希望确保实际提供所有字段名称的翻译。

或者，如何在 Elasticsearch 中使用合理的字段名称 :-) ？