快捷導(dǎo)航

pandas采樣的實(shí)現(xiàn)方法

更新時(shí)間：2024年12月31日 10:47:20 作者：frostjsy

pandas提供了多種方法進(jìn)行隨機(jī)采樣,包括指定數(shù)量、按比例、設(shè)置隨機(jī)種子、有無放回抽樣及按列和分層抽樣,下面就來介紹一下,感興趣的可以了解一下

基本用法：DataFrame 的 sample 方法
示例數(shù)據(jù)
示例 1：隨機(jī)抽取指定數(shù)量的行
示例 2：按比例隨機(jī)抽樣
示例 3：隨機(jī)抽樣時(shí)指定隨機(jī)種子
示例 4：按行或按列隨機(jī)抽樣
示例 5：返回是否放回抽樣
示例 6：按某列進(jìn)行分層隨機(jī)抽樣
總結(jié)

在數(shù)據(jù)分析和處理過程中，常常需要對(duì)數(shù)據(jù)進(jìn)行隨機(jī)采樣，以便獲取具有代表性的小樣本或進(jìn)行數(shù)據(jù)拆分。pandas 提供了非常方便的方法來實(shí)現(xiàn)隨機(jī)采樣。

基本用法：DataFrame 的 sample 方法

pandas 的 DataFrame 提供了 sample 方法用于隨機(jī)采樣，下面說明其基本用法和常見參數(shù)：

示例數(shù)據(jù)

首先，創(chuàng)建一個(gè)示例 DataFrame：

import pandas as pd

# 創(chuàng)建一個(gè)示例 DataFrame
data = {
    'name': ['Alice', 'Bob', 'Charlie', 'David', 'Eve', 'Frank', 'Grace', 'Hannah'],
    'age': [25, 30, 35, 40, 45, 50, 55, 60],
    'city': ['New York', 'Los Angeles', 'Chicago', 'Houston', 'Phoenix', 'Philadelphia', 'San Antonio', 'San Diego']
}
df = pd.DataFrame(data)
print(df)

示例 1：隨機(jī)抽取指定數(shù)量的行

可以使用 n 參數(shù)指定抽取的行數(shù)：

# 隨機(jī)抽取 3 行數(shù)據(jù)
sampled_df = df.sample(n=3)

print(sampled_df)

示例 2：按比例隨機(jī)抽樣

可以使用 frac 參數(shù)指定抽樣的比例，例如 frac=0.5 表示隨機(jī)抽取 50% 的行數(shù)：

# 隨機(jī)抽取 50% 的行
sampled_df = df.sample(frac=0.5)

print(sampled_df)

示例 3：隨機(jī)抽樣時(shí)指定隨機(jī)種子

為了保證每次抽樣結(jié)果相同，可以使用 random_state 參數(shù)指定隨機(jī)種子：

# 隨機(jī)抽取 3 行數(shù)據(jù)，并指定隨機(jī)種子
sampled_df = df.sample(n=3, random_state=1)

print(sampled_df)

示例 4：按行或按列隨機(jī)抽樣

默認(rèn)情況下，sample 是按行抽樣（axis=0），也可以設(shè)置 axis=1 來按列進(jìn)行抽樣：

# 按列隨機(jī)抽取 2 列
sampled_df = df.sample(n=2, axis=1)

print(sampled_df)

示例 5：返回是否放回抽樣

默認(rèn)情況下，sample 是不放回抽樣，即一個(gè)樣本只會(huì)被抽取一次。可以設(shè)置 replace=True 來進(jìn)行有放回抽樣：

# 進(jìn)行有放回抽樣，隨機(jī)抽取 10 個(gè)樣本
sampled_df = df.sample(n=10, replace=True, random_state=1)

print(sampled_df)

示例 6：按某列進(jìn)行分層隨機(jī)抽樣

有時(shí)需要按某列的值進(jìn)行分層隨機(jī)抽樣，可以使用 groupby 和 apply 方法結(jié)合 sample 來實(shí)現(xiàn)：

# 按 'city' 列進(jìn)行分層隨機(jī)抽樣，每個(gè)城市隨機(jī)抽取 1 行
sampled_df = df.groupby('city').apply(lambda x: x.sample(n=1, random_state=1)).reset_index(drop=True)

print(sampled_df)

總結(jié)

pandas 提供了豐富的參數(shù)和功能，讓 sample 方法能夠滿足各種隨機(jī)抽樣的需求，包括指定抽樣數(shù)量、按比例抽樣、設(shè)置隨機(jī)種子、有無放回抽樣以及按列和分層抽樣。這些功能在數(shù)據(jù)分析和處理過程中非常有用，有助于快速獲取具有代表性的小樣本進(jìn)行分析。

到此這篇關(guān)于pandas采樣的實(shí)現(xiàn)方法的文章就介紹到這了,更多相關(guān)pandas采樣內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: