Skip to content

Commit ea891b2

Browse files
Use field_name if available
1 parent f6bdd1d commit ea891b2

File tree

1 file changed

+9
-6
lines changed

1 file changed

+9
-6
lines changed

python/pyarrow/pandas_compat.py

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -648,12 +648,15 @@ def _add_any_metadata(table, pandas_metadata):
648648
# Add time zones
649649
for i, col_meta in enumerate(pandas_metadata['columns']):
650650

651-
raw_name = col_meta['name']
652-
if i >= n_columns:
653-
# index columns
654-
raw_name = index_columns[i - n_columns]
655-
if raw_name is None:
656-
raw_name = 'None'
651+
raw_name = col_meta.get('field_name')
652+
if not raw_name:
653+
# deal with metadata written with arrow < 0.8
654+
raw_name = col_meta['name']
655+
if i >= n_columns:
656+
# index columns
657+
raw_name = index_columns[i - n_columns]
658+
if raw_name is None:
659+
raw_name = 'None'
657660

658661
idx = schema.get_field_index(raw_name)
659662
if idx != -1:

0 commit comments

Comments
 (0)