Как читать указанное количество строк из csv файла в python 3?

Пользователь

от autumn_parker , в категории: Другие , 7 месяцев назад

Как читать указанное количество строк из csv файла в python 3?

Facebook Vk Ok Twitter LinkedIn Telegram Whatsapp

2 ответа

Пользователь

от ottilie.farrell , 7 месяцев назад

@autumn_parker 

В Python 3 можно использовать модуль csv для чтения csv-файлов. Вот пример кода, который считывает указанное количество строк из csv-файла:

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
import csv

def read_csv(path, num_rows):
    with open(path, 'r') as file:
        reader = csv.reader(file)
        rows = []
        for i, row in enumerate(reader):
            if i >= num_rows:
                break
            rows.append(row)
    return rows

# Пример использования
path = 'file.csv'
num_rows = 10
data = read_csv(path, num_rows)
print(data)


В этом примере функция read_csv принимает путь к файлу и количество строк, которые нужно прочитать. Затем она открывает файл и использует цикл для итерации по строкам до достижения указанного количества строк. Каждая строка добавляется в список rows. В конце функция возвращает этот список.


Пример использования показывает, как прочитать первые 10 строк из файла "file.csv" и вывести их на экран.

Пользователь

от velma , 2 месяца назад

@autumn_parker 

Этот код считает csv-файл как текст, что означает, что знаки новой строки внутри кавычек будут обработаны неверно. Чтобы избежать этой проблемы, рекомендуется использовать библиотеку pandas для более надежного чтения csv-файлов. Вот пример кода с использованием pandas:

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
import pandas as pd

def read_csv(path, num_rows):
    df = pd.read_csv(path, nrows=num_rows)
    return df

# Пример использования
path = 'file.csv'
num_rows = 10
data = read_csv(path, num_rows)
print(data)


Этот код считывает первые 10 строк из файла "file.csv" с использованием библиотеки pandas и выводит их в виде таблицы данных. При использовании pandas вы пользуетесь библиотекой, которая обрабатывает строки и столбцы данных в более удобной форме, чем стандартная библиотека csv.