Python CSV 数据提取与类型转换：提取特定列并转换为浮点数

Python 提取 CSV 数据并转换为浮点数

本文将介绍如何使用 Python 提取 CSV 文件中特定列的数据，并将它们转换为浮点数。

代码示例：

import csv

def extract_and_convert(filename):
    extracted_data = []
    
    with open(filename, 'r') as file:
        reader = csv.reader(file)
        for row in reader:
            extracted_row = []
            for i in range(3, 11):
                extracted_row.append(float(row[i]))
            extracted_data.append(extracted_row)
    
    return extracted_data

# 调用示例
filename = 'data.csv'
# 替换为您的实际数据文件路径
data = extract_and_convert(filename)
print(data)

代码解释：

导入 csv 模块： import csv 用于处理 CSV 文件。
定义函数 extract_and_convert： 该函数接收 CSV 文件路径作为参数，并返回提取的浮点数数据。
打开文件： 使用 with open(filename, 'r') as file: 打开 CSV 文件进行读取。
创建 CSV 读取器： reader = csv.reader(file) 创建一个 CSV 读取器，用于逐行读取文件内容。
遍历每一行： 使用 for row in reader: 循环遍历文件中的每一行。
提取并转换数据： 使用嵌套循环，提取第 4 列到第 11 列的数据，并使用 float(row[i]) 将其转换为浮点数。
存储提取数据： 将转换后的浮点数数据存储到 extracted_data 列表中。
返回提取数据： 函数返回 extracted_data 列表，包含所有提取的浮点数数据。
调用函数： 使用 filename = 'data.csv' 指定数据文件路径，并使用 data = extract_and_convert(filename) 调用函数。
输出结果： 使用 print(data) 打印提取的浮点数数据。

注意事项：

请确保 data.csv 文件存在，并将代码中的文件路径替换为您的实际文件路径。
确保数据文件中第 4 列到第 11 列包含可以转换为浮点数的数据。
您可以根据需要对返回的二维列表 data 进行进一步处理。

希望本代码示例能够帮助您提取 CSV 文件中的特定列数据并将其转换为浮点数。