Как определить количество столбцов в файле CSV в Python

Question

У меня есть CSV-файл, который в начале имеет 2 столбца, а в некоторой строке 4 столбца. Я хочу использовать только 2 столбца и игнорировать остальные.

мой код:

df = pd.read_csv(path,header=None,nrows=NUM_OF_CONF_ROWS,skiprows=2)

ошибка:

Ошибка токенизации данных. Ошибка С: ожидается 2 поля в строке 17, пила 4

пример данных:

 1,2
 2,4
 ...
 1,2,3,4
 1,2

Answer 1

Нет такой вещи, что в начале у вас есть только 2 столбца и всего 4 столбца из-за наличия непустых значений в этом столбце на более высоких индексах. Это означает, что ваш CSV-файл действительно имеет 4 столбца для начала.

Если вам просто нужны первые два столбца, вы можете «отбросить» остальные, используя метод df.columns.isin(), например:

df = df.loc[df.columns.isin('place column names or index in a list and put it here')]

И вы получите 2 нужных столбца.

Надеюсь, это поможет:))

Answer 2

Вы можете использовать опцию usecols, чтобы указать, какие столбцы вы хотите использовать:

df = pd.read_csv(path,header=None,nrows=NUM_OF_CONF_ROWS,skiprows=2, usecols = [0 1])