pandas - 如何将pandas数据导出到elasticsearch？

Question

可以使用 elasticsearch-py 将 pandas dataframe 数据导出到 elasticsearch。例如，这里有一些代码:

https://www.analyticsvidhya.com/blog/2017/05/beginners-guide-to-data-exploration-using-elastic-search-and-kibana/

to_excel、to_csv、to_sql等类似方法还有很多。

是否有to_elastic方法？如果没有，我应该在哪里申请？

Answer 1

以下脚本适用于本地主机:

import numpy as np
import pandas as pd

df = pd.DataFrame(np.random.randint(0,100,size=(100, 4)), columns=list('ABCD'))

INDEX="dataframe"
TYPE= "record"

def rec_to_actions(df):
    import json
    for record in df.to_dict(orient="records"):
        yield ('{ "index" : { "_index" : "%s", "_type" : "%s" }}'% (INDEX, TYPE))
        yield (json.dumps(record, default=int))

from elasticsearch import Elasticsearch
e = Elasticsearch() # no args, connect to localhost:9200
if not e.indices.exists(INDEX):
    raise RuntimeError('index does not exists, use `curl -X PUT "localhost:9200/%s"` and try again'%INDEX)

r = e.bulk(rec_to_actions(df)) # return a dict

print(not r["errors"])

使用 curl -g 'http://localhost:9200/dataframe/_search?q=A:[29%20TO%2039]' 验证

可以添加许多小东西以满足不同的需求，但主要的还是有。