根据 CSV 文件行中的值确定数据类型

Question

我编写了一个 Python 程序，它读取 CSV 文件并将它们作为 MySQL 的插入语句输出。现在我想要一个能够使用列标题为每个新表形成CREATE 语句的附加程序。由于 MySQL 格式，我需要能够确定列中每个值的类型，类似于 shell 中的 type() 函数。我的想法是在列标题之后读取 CSV 文件的第二行，并提取由 , 分隔的值（即每列中的一个值）。然后获取该值并通过type() 函数运行它并返回int, str, float, etc.。稍后我将使用type() 返回的值执行一些if statements，以将正确的数据类型附加到CREATE 语句的列标题的末尾。

到目前为止，我已经设法将每个值从第一行的编译列表中提取出来并打印出它的 type()。唯一的问题是列表中的所有内容都被视为字符串，因此唯一的返回是<class 'str'>。如果我知道它是什么类型的值，我将能够从字符串转换为正确的类型，但这违背了程序的整个目的。我确信有一种更好的方法可以做到这一点，它不会将第一行作为字符串列表吐出，但我不知道该怎么做。以下是我目前所拥有的以及如何继续的想法：

import csv, os

path = 'C:/Users/user/Desktop/file/test/'
for file in os.listdir(path):
    if file.endswith('.csv'):
        with open(path +file) as inFile:
            with open(path + file[:-4] + ".txt", "w") as outFile:
                csvFile = csv.reader(inFile)
                columnHeader = next(csvFile)
                firstRow = next(csvFile)
                i = 0
                for value in firstRow:
                    valueType = (type(value))
                    for header in columnHeader:
                        if valueType = class 'str': # Don't think this formatting is correct
                            columnHeader.append(' varchar (255)')
                        if valueType = class 'int':
                            columnHeader.append(' int')
                        if valueType = class 'float':
                            columnHeader.append(' float')

我也研究过使用 this 库，但更愿意自己做。

我选择的方法是否合理/可行？你能想到什么更好的方法来做到这一点吗？如果有，怎么做？

Answer 1

这是简单的方法：

def find_type(a):
    try:
        var_type = type(int(a))
    except ValueError:
        try:
            var_type = type(float(a))
        except ValueError:
            var_type = type(a)
    return var_type

a = ['123123', '11.21', 'Some Bank', '11/2/1995']

print([find_type(x) for x in a])
#[<class 'int'>, <class 'float'>, <class 'str'>, <class 'str'>]

我很确定还有更优雅的方法。

Answer 2

这是区分 float 和 int 并假设其他所有内容都是字符串的一种可能性。如果该值有小数点，则它会尝试将其转换为浮点数，如果没有则尝试将其转换为 int。如果任一转换失败，则将值保留为字符串。

if '.' in val:
    try:
        val = float(val)
    except ValueError:
        pass
else:
    try:
        val = int(val)
    except ValueError:
        pass