Update es_pandas.py

fuyb1992 · Dec 9, 2019 · cb7355a · cb7355a
1 parent 1be9a47
commit cb7355a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/es_pandas/es_pandas.py b/es_pandas/es_pandas.py
@@ -107,7 +107,7 @@ def to_pandas(self, index, query_rule={'query': {'match_all': {}}}, heads=[], dt
         for _ in progressbar.progressbar(range(0, count)):
             mes = next(anl)
             for head in heads:
-                df_li[head].append(mes['_source'][head])
+                df_li[head].append(mes['_source'][head] if head in mes['_source'] else np.nan)
             df_li[self.id_col].append(mes['_id'])
 
         return pd.DataFrame(df_li).set_index(self.id_col).astype(dtypes)