База данных csv python
У меня имеется csv бд с n количеством строк, как можно получить рандомные строки из нее с помощью python?
Источник: Stack Overflow на русском
Ответы
Вот так:
import csv
import random
def get_random_string(num):
with open('test.txt') as f:
reader = csv.reader(f)
return random.sample(list(reader), k=num)
num - количество строк.
import pandas as pd
def get_random_sample(sample_len: int):
df = pd.read_csv('test.txt') # Чтение csv файла с помощью pandas
return df.sample(n=sample_len)