python中sample函数的介绍与使用

目录

• 1 sample(序列a，n)
• 2 random sample()
• 补充：Pandas-sample随机抽样实例
• 总结

1 sample(序列a，n)

sample(序列a，n)

功能：从序列a中随机抽取n个元素，并将n个元素生以list形式返回。

例：

from random import randint, sample
date = [randint(10,20) for _ in range(10)]
c = sample(date, 5)
print(c)

# 输出：[12, 17, 10, 12, 17]

randint(10,20) for _ in range(10)：从10~20间随机抽取10个数；

2 random sample()

sample()方法返回一个列表，其中从序列中随机选择指定数量的项目。

注意：此方法不会更改原始顺序。

实例

返回一个包含列表中任何两项的列表：

import random
mylist = ["apple", "banana", "cherry"]
print(random.sample(mylist, k=2))

语法

random.sample(sequence, k)

补充：Pandas-sample随机抽样实例

按行抽取

import pandas as pd
#创建DataFrame数据
info = pd.DataFrame({'name': ['李四', '王五', '小贝', '虞姬'],
     'age': [18, 19, 20, 18],
     'sex': ['男', '男', '女','女' ],
     'class': ['5班', '4班', '5班','3班' ]})
#默认随机选择两行
print(info.sample(n=2))

结果

按列抽取

import pandas as pd
#创建DataFrame数据
info = pd.DataFrame({'name': ['李四', '王五', '小贝', '虞姬'],
     'age': [18, 19, 20, 18],
     'sex': ['男', '男', '女','女' ],
     'class': ['5班', '4班', '5班','3班' ]})
#随机选择两列
print(info.sample(n=2,axis=1))

结果

总结