在 numpy 中使用样本和种子

Question

die = pd.DataFrame([1, 2, 3, 4, 5, 6])

sum_of_dice = die.sample(n=2, replace=True).sum().loc[0]

print (sum_of_dice)

有人可以解释一下.sum().loc[0] 在这里做什么吗？

Answer 1

打印中间步骤以获得想法总是有用的。

sum 计算每列数据帧的总和。

https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.DataFrame.sum.html

loc 选择一组行/列。

https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.DataFrame.loc.html

sum 返回一个包含一个元素的数据帧，但由于我们需要整数 dtype 中的和而不是数据帧，所以我们使用 loc 来获取第一个元素。

import pandas as pd

die = pd.DataFrame([1, 2, 3, 4, 5, 6])

sum_of_dice = die.sample(n=2, replace=True)

print(sum_of_dice)

sum_of_dice = sum_of_dice.sum()

print('---')
print (sum_of_dice)

sum_of_dice = sum_of_dice.loc[0]
print('---')
print (sum_of_dice)

  0
4  5
0  1
---
0    6
dtype: int64
---
6