下面是一个示例代码,该代码使用Python的csv模块遍历CSV文件,并确定每列的数据类型。
import csv
def determine_data_types(csv_file):
with open(csv_file, 'r') as file:
reader = csv.reader(file)
header = next(reader) # 获取CSV文件的首行,即列名
num_columns = len(header)
data_types = [''] * num_columns # 初始化一个与列数相同的空列表,用于存储每列的数据类型
for row in reader:
for i in range(num_columns):
cell_value = row[i].strip() # 去除单元格中的空格
# 判断数据类型
if cell_value.isdigit():
if data_types[i] != 'str':
data_types[i] = 'int'
elif cell_value.replace('.', '', 1).isdigit():
if data_types[i] != 'str' and data_types[i] != 'int':
data_types[i] = 'float'
else:
data_types[i] = 'str'
return data_types
# 示例用法
csv_file = 'data.csv'
data_types = determine_data_types(csv_file)
print(data_types)
请注意,此示例代码假设CSV文件中的每列都包含相同类型的数据。如果某一列的数据类型不统一,则可能需要更复杂的逻辑来确定每列的数据类型。
上一篇:遍历CSV文件中的行
下一篇:遍历CSV直到特定行