[ARROW-3651] [Python] Datetimes from non-DateTimeIndex cannot be deserialized

• Type: Bug
• Status: Resolved
• Priority: Major
• Resolution: Fixed
• Affects Version/s: 0.11.1
• Fix Version/s: 0.15.0
• Component/s: Python
• Labels: parquet pull-request-available
• External issue URL: https://github.com/apache/arrow/issues/19958
• Assignee: Wes McKinney
• Reporter: Armin Berres
• Created: 30/Oct/18 09:23
• Updated: 11/Jan/23 07:28
• Resolved: 10/Sep/19 01:23

Description

Given an index which contains datetimes but is no DateTimeIndex writing the file works but reading back fails.

df = pd.DataFrame(1, index=pd.MultiIndex.from_arrays([[1,2],[3,4]]), columns=[pd.to_datetime("2018/01/01")])

# columns index is no DateTimeIndex anymore
df = df.reset_index().set_index(['level_0', 'level_1'])

table = pa.Table.from_pandas(df)
pq.write_table(table, 'test.parquet')

pq.read_pandas('test.parquet').to_pandas()

results in

KeyError                                  Traceback (most recent call last)
~/venv/mpptool/lib/python3.7/site-packages/pyarrow/pandas_compat.py in _pandas_type_to_numpy_type(pandas_type)
    676     try:
--> 677         return _pandas_logical_type_map[pandas_type]
    678     except KeyError:

KeyError: 'datetime'

The created schema:

2018-01-01 00:00:00: int64
level_0: int64
level_1: int64
metadata
--------
{b'pandas': b'{"index_columns": ["level_0", "level_1"], "column_indexes": [{"n'
            b'ame": null, "field_name": null, "pandas_type": "datetime", "nump'
            b'y_type": "object", "metadata": null}], "columns": [{"name": "201'
            b'8-01-01 00:00:00", "field_name": "2018-01-01 00:00:00", "pandas_'
            b'type": "int64", "numpy_type": "int64", "metadata": null}, {"name'
            b'": "level_0", "field_name": "level_0", "pandas_type": "int64", "'
            b'numpy_type": "int64", "metadata": null}, {"name": "level_1", "fi'
            b'eld_name": "level_1", "pandas_type": "int64", "numpy_type": "int'
            b'64", "metadata": null}], "pandas_version": "0.23.4"}'}

Issue Links

• links to: GitHub Pull Request #5311