import pandas as pd
from elasticsearch import Elasticsearch
fname="FILE_NAME.tsv"
df_tsv = pd.read_csv(fname, sep='\t', dtype=str)
es = Elasticsearch("http://{IP}:9200")
search_list = df_tsv['検索対象'].tolist()
search_count_res = []
##############################
for child in search_list:
es_count_res = es.count(
index="one_year_comment",
body={
"query": {
"query_string": {
"default_field" : "comment",
"query":'\"'+child+'\"'
}
}
})
search_count_res.append(es_count_res["count"])
##############################
df_tsv['search_count'] = search_count_res
# print( df_tsv[["検索対象","search_count"]] )
df_tsv.to_csv('OUTPUT_SERACH_COMMENT.tsv', sep='\t')