Skip to content

Commit 796eaa3

Browse files
make change back compat
1 parent 57d8686 commit 796eaa3

File tree

1 file changed

+10
-6
lines changed

1 file changed

+10
-6
lines changed

python/pyarrow/pandas_compat.py

+10-6
Original file line numberDiff line numberDiff line change
@@ -191,8 +191,9 @@ def get_column_metadata(column, name, arrow_type, field_name):
191191
}
192192

193193

194-
def construct_metadata(columns_to_convert, df, column_names, column_field_names,
195-
index_levels, index_descriptors, preserve_index, types):
194+
def construct_metadata(columns_to_convert, df, column_names, index_levels,
195+
index_descriptors, preserve_index, types,
196+
column_field_names=None):
196197
"""Returns a dictionary containing enough metadata to reconstruct a pandas
197198
DataFrame as an Arrow Table, including index columns.
198199
@@ -211,6 +212,9 @@ def construct_metadata(columns_to_convert, df, column_names, column_field_names,
211212
-------
212213
dict
213214
"""
215+
if column_field_names is None:
216+
column_field_names = column_names
217+
214218
num_serialized_index_levels = len([descr for descr in index_descriptors
215219
if not isinstance(descr, dict)])
216220
# Use ntypes instead of Python shorthand notation [:-len(x)] as [:-0]
@@ -565,8 +569,8 @@ def dataframe_to_types(df, preserve_index, columns=None):
565569
types.append(type_)
566570

567571
metadata = construct_metadata(
568-
columns_to_convert, df, column_names, column_field_names, index_columns,
569-
index_descriptors, preserve_index, types
572+
columns_to_convert, df, column_names, index_columns, index_descriptors,
573+
preserve_index, types, column_field_names=column_field_names
570574
)
571575

572576
return all_names, types, metadata
@@ -649,8 +653,8 @@ def _can_definitely_zero_copy(arr):
649653
schema = pa.schema(fields)
650654

651655
pandas_metadata = construct_metadata(
652-
columns_to_convert, df, column_names, column_field_names, index_columns,
653-
index_descriptors, preserve_index, types
656+
columns_to_convert, df, column_names, index_columns, index_descriptors,
657+
preserve_index, types, column_field_names=column_field_names
654658
)
655659
metadata = deepcopy(schema.metadata) if schema.metadata else dict()
656660
metadata.update(pandas_metadata)

0 commit comments

Comments
 (0)