Как сдвинуть заголовки столбцов в pandas

Question

У меня есть .txt-файлы, которые я читаю с помощью pandas, а строка заголовка начинается с '~ A'. Мне нужно игнорировать «~ A», а следующий заголовок соответствует данным в первом столбце. Благодаря!

Answer 1

Вы можете сделать это:

import pandas as pd 
data = pd.read_csv("./test.txt", names=[ 'A', 'B' ], skiprows=1)  
print(data) 

и выход для ввода:

~A, A, B 
1, 2 
3, 4 

является:

c:\Temp\python>python test.py 

    A B 
0 1 2 
1 3 4 

Вы должны назвать столбцы самостоятельно, но при условии, что файл Кажется, что он неправильный. Думаю, это не так уж плохо.

Если строки заголовка не являются одинаковыми во всех файлах, то вы можете просто читать их в Python:

import pandas as pd; 

# read first line 
with open("./test.txt") as myfile: 
    headRow = next(myfile) 

# read column names  
columns = [x.strip() for x in headRow.split(',')] 

# process by pandas 
data = pd.read_csv("./test.txt", names=columns[1:], skiprows=1) 

print(data);