From b10f7286b7e1131ad013c437fef7a1ac51c7705b Mon Sep 17 00:00:00 2001 From: Kyle Barron Date: Mon, 12 Nov 2018 14:52:14 -0500 Subject: [PATCH] More helpful Stata object dtype error. (#23572) --- pandas/io/stata.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/pandas/io/stata.py b/pandas/io/stata.py index 66e996075f1ed..8affc90773d9a 100644 --- a/pandas/io/stata.py +++ b/pandas/io/stata.py @@ -466,6 +466,12 @@ def parse_dates_safe(dates, delta=False, year=False, days=False): """ +general_object_array_error = """ +Writing non-string object columns is not supported. Column '%s' does not +satisfy this restriction. +""" + + class PossiblePrecisionLoss(Warning): pass @@ -1868,7 +1874,7 @@ def _dtype_to_default_stata_fmt(dtype, column, dta_version=114, inferred_dtype = infer_dtype(column.dropna()) if not (inferred_dtype in ('string', 'unicode') or len(column) == 0): - raise ValueError('Writing general object arrays is not supported') + raise ValueError(general_object_array_error % column.name) itemsize = max_len_string_array(ensure_object(column.values)) if itemsize > max_str_len: if dta_version >= 117: