python - 如何在Python中仅选择数值数据类型并将它们保存在数组中？

Question

我想从 csv 中获取所有数字字段，并将这些字段存储在数组中，以便我可以执行数学运算。我可以获取数据类型但无法限制。我对 python 脚本非常陌生，请帮忙

编辑:我添加了一个示例行

所以这里F1和F3是数值字段。所以我想将这两个字段名称保留在数组变量中FieldNames=["F1","F3"]

import csv
import pandas as pd
import numpy as np

data = pd.read_csv(r'C:\Users\spanda031\Downloads\test_19.csv')
print(data.dtypes)
with open(r'C:\Users\spanda031\Downloads\test_19.csv') as f:
    d_reader = csv.DictReader(f)

    #get fieldnames from DictReader object and store in list
    headers = d_reader.fieldnames
    print(headers)
    for line in headers:
        #print value in MyCol1 for each row
        print(line)        
        v3=np.array(data[line])

Answer 1

select_dtypes

您可以使用 np.number 或如文档中所示，'number' 选择所有数字系列:

# read csv file
df = pd.read_csv('file.csv')

# subset dataframe to include only numeric columns
df = df.select_dtypes(include='number')

# get column labels in array
cols = df.columns.values

# extract NumPy array from dataframe
arr = df.values

请注意，不需要 csv 模块，因为 Pandas 可以通过 pd.read_csv 读取 csv 文件。