выбрать несколько строк по индексу из dask dataframe?

Question

df = dd.read_csv('csv',usecols=fields,skip_blank_lines=True)
len(df.iloc[0:5])

Приведенный выше код повышает

AttributeError: 'DataFrame' object has no attribute 'iloc'

пробовал ix loc, но не мог выбрать строки на основе индекса

Answer 1

Один из обходных путей - создать индекс в виде столбца, т.е. df_index, в вашем CSV-файле и использовать его следующим образом:

selection = (df[ df['df_index'].isin( list_of_indexes ) ]).compute()

Answer 2

Dask.dataframe не поддерживает iloc. Обычно довольно сложно получить доступ к какой-либо конкретной строке в CSV-файле, не прочитав все это в память.

Однако, если вам нужны только несколько строк в верхней части, я рекомендую использовать метод .head()

>>> df.head()