在 CSV 文件中存储和检索布尔值的便捷方法是什么

Question

如果我使用 CSV 模块存储布尔值，它会被 str() 函数转换为字符串 True 或 False 。但是，当我加载这些值时，False 字符串的计算结果为 True，因为它是一个非空字符串。

我可以通过在读取时使用 IF 语句手动检查字符串来查看字符串是什么来解决这个问题，但这有点不太优雅。有更好的想法吗？或者这只是编程世界中的其中之一？

Answer 1

我不确定回答你自己的问题是否是一种不好的形式，但这是我提出的解决方案。它基本上包括将我正在谈论的讨厌的 IF 语句分离到一个函数中。

def setyesNo(value):
    if value: return 'Yes'
    else: return 'No'

def checkYesNo(text):
    if text == 'Yes': return True
    else: return False

然后在我的 dictWriter 中执行此操作。

for item in mylist:
    writer.writerow( {'Is Cool' : setYesNo(item.is_cool),
                      .....
                      })

并在 dictReader 中。

for line in reader:
    item MyObject(is_Cool=checkYesNo(line['Is Cool']),
                  .....
                  )
    mylist.append(item)

Answer 2

使用 int() 函数将布尔值转换为 int 值，然后存储它们。话虽这么说，伊莱-本德斯基的上述评论值得注意。

Answer 3

几乎总是有一种更简单（而且通常更方便）的方法，而且大多数时候它已经包含在 Python 令人惊叹的标准库中。

在这种情况下，您可以只使用 True 和 False，然后使用 ast 的 literal_eval 方法解析它们。

假设我们有一个名为 test.csv 的文件，其中包含以下内容：

nonetype,integer,boolean
None,123,False

我们还有一个 test.py 脚本来解析该文件：

import ast
import csv

with open('test.csv', 'r') as csvfile:
    reader = csv.DictReader(csvfile)
    for row in reader:
        nonetype = ast.literal_eval(row['nonetype'])
        integer = ast.literal_eval(row['integer'])
        boolean = ast.literal_eval(row['boolean'])
        print(type(nonetype), type(integer), type(boolean))

当我们运行该脚本时，我们可以看到这些值按预期进行解析。

$ python3 test.py
<class 'NoneType'> <class 'int'> <class 'bool'>

即使输入来自不受信任的来源，此方法也可以安全使用，更多信息 文档。

Answer 4

在 CSV 文件中存储布尔值的方法

• 字符串：两个常见的选择是true和false、True和False，但我也见过是和否。
• 整数：0 或 1
• 浮点数：0.0 或 1.0

让我们比较一下各自的优点/缺点：

• 字符串：
  • + 人类可以阅读
  • - CSV 读取器会将其作为字符串，并且当对其应用 bool 时，两者的计算结果都会为“true”
• 整数：
  • + CSV 读者可能会看到此列是整数，并且 bool(0) 计算结果为 false。
  • + 更节省空间
  • - 不太清楚它是布尔值
• 浮点数：
  • + CSV 读者可能会看到此列是整数，并且 bool(0.0) 计算结果为 false。
  • - 不太清楚它是布尔值
  • + 可能为 null（如 NaN）

Pandas CSV 阅读器显示所描述的行为。

将 Bool 字符串转换为 Bool 值

查看 mpu。 string.str2bool：

>>> str2bool('True')
True
>>> str2bool('1')
True
>>> str2bool('0')
False

其实现如下：

def str2bool(string_, default='raise'):
    """
    Convert a string to a bool.

    Parameters
    ----------
    string_ : str
    default : {'raise', False}
        Default behaviour if none of the "true" strings is detected.

    Returns
    -------
    boolean : bool

    Examples
    --------
    >>> str2bool('True')
    True
    >>> str2bool('1')
    True
    >>> str2bool('0')
    False
    """
    true = ['true', 't', '1', 'y', 'yes', 'enabled', 'enable', 'on']
    false = ['false', 'f', '0', 'n', 'no', 'disabled', 'disable', 'off']
    if string_.lower() in true:
        return True
    elif string_.lower() in false or (not default):
        return False
    else:
        raise ValueError('The value \'{}\' cannot be mapped to boolean.'
                         .format(string_))