python - Pandas 使用 Numpy 作为随机数生成器吗？

Question

我想要获得可重复的数据样本。一项快速实验表明，numpy.random.seed确实影响pandas.DataFrame.sample ，但没有记录。

有人知道吗

我尝试了什么

我运行了以下几次并且总是得到相同的结果

#!/usr/bin/env python

import pandas as pd
import numpy as np


df = pd.DataFrame([(1, 2, 1),
                   (1, 2, 2),
                   (1, 2, 3),
                   (4, 1, 612),
                   (4, 1, 612),
                   (4, 1, 1),
                   (3, 2, 1),
                   ],
                  columns=['groupid', 'a', 'b'],
                  index=['India', 'France', 'England', 'Germany', 'UK', 'USA',
                         'Indonesia'])
np.random.seed(0)
print(df.sample(n=1))
print(df.sample(n=1))
print(df.sample(n=1))
print(df.sample(n=1))
print(df.sample(n=1))

这给出:

• 印度尼西亚
• 法国
• 印度尼西亚
• 美国
• 英格兰

Answer 1

pandas 使用 _random_state 函数返回 np.random.RandomState link :

def _random_state(state=None):
    """
    Helper function for processing random_state arguments.
    Parameters
    ----------
    state : int, np.random.RandomState, None.
        If receives an int, passes to np.random.RandomState() as seed.
        If receives an np.random.RandomState object, just returns object.
        If receives `None`, returns np.random.
        If receives anything else, raises an informative ValueError.
        Default None.
    Returns
    -------
    np.random.RandomState
    """

    if types.is_integer(state):
        return np.random.RandomState(state)
    elif isinstance(state, np.random.RandomState):
        return state
    elif state is None:
        return np.random
    else:
        raise ValueError("random_state must be an integer, a numpy "
                         "RandomState, or None")

以及 sample称为这个函数:

    # Process random_state argument
    rs = com._random_state(random_state)