Сопоставление значений строки одного CSV со значениями заголовка другого CSV и запись нового файла CSV в python

Question

import numpy as np
import pandas as pd
import csv

data=np.loadtxt(open("ff.csv","rb"),delimiter=",",dtype=np.str)
print(data)

data1=np.loadtxt(open("x.csv","rb"),delimiter=",",dtype=np.str)
print(data1)

for i in range(1,len(data)):

    for j in range(len(data1[0])):

        if data[i] == data1[0][j]:
            print(data[i],",",data1[0][j])

            column = pd.read_csv("x.csv",usecols=[data1[0][j]])
            print(column)

Это первое имя данных csv в строке

здесь данные в столбцах

хотите сопоставить имя строки с имя столбца и сохраните его в новый CSV в соответствии с соответствием со строкой

выведите то, что я хочу здесь

Answer 1

Pandas - это полезный инструмент: вы можете перетащить имена столбцов в список, передать этот список на второй лист данных в виде подмножества столбцов, которые нужно извлечь, а затем сохранить этот фрейм данных в новый файл.

data=pd.read_csv("ff.csv", header = 0)

data1=pd.read_csv("x.csv", header = 0)

column_list = data['Name'].to_list()
column_list.append('class')

data1[column_list].to_csv("newfile.csv",index=False)