From 86cc239181bb2c55b7a95c9ea52403ccde01ec55 Mon Sep 17 00:00:00 2001 From: Mark O'Leary Date: Thu, 26 Jul 2012 16:21:45 -0400 Subject: [PATCH] ENH: add more possible bool values to read_csv #1295 --- pandas/src/inference.pyx | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/pandas/src/inference.pyx b/pandas/src/inference.pyx index c71e08ae5afa5..d7640b1719f07 100644 --- a/pandas/src/inference.pyx +++ b/pandas/src/inference.pyx @@ -654,9 +654,11 @@ def maybe_convert_bool(ndarray[object] arr): for i from 0 <= i < n: val = arr[i] - if val == 'True' or type(val) == bool and val: + true_vals = ('True', 'TRUE', 'true', 'Yes', 'YES', 'yes') + false_vals = ('False', 'FALSE', 'false', 'No', 'NO', 'no') + if val in true_vals or type(val) == bool and val: result[i] = 1 - elif val == 'False' or type(val) == bool and not val: + elif val in false_vals or type(val) == bool and not val: result[i] = 0 else: return arr