python - 在 python 中取一个没有 numpy 或 pandas 的数组样本

Question

我正在尝试创建一种“pythonic”方式，在 python 中获取一个非常大的数组的一小部分。

我目前正在接收一个包含 58 列和 4960 行的 csv，代码如下:

def import_normal_csv(file):
    # Create blank array
    results = []
    # Open file
    with open(file) as csvfile:
        # read in file changing values to floats
        reader = csv.reader(csvfile, quoting=csv.QUOTE_NONNUMERIC)
        for row in reader:
            results.append(row)
    return results

def main():
    print(" Working SPAM Dataset... ")
    # Create a raw data array without numpy
    spam_raw_data = import_normal_csv('spam.csv')

    # CREATE SUBSET OF SPAM_RAW_DATA HERE

    random.shuffle(spam_raw_data)

我已经看到使用 numpy 或 pandas 执行此操作的各种方法，但我想在没有这些库的情况下自然地执行此操作。而不是我庞大的数组，我怎么能只取...500 行(或某个明显少于近 5000 行的任意数字)？

Answer 1

你可以使用内置的random库，例如:

import random

random.sample(data, 500)

这将为您提供一个包含 500 个 list 的 list，每个代表一行。