表达式1 if 条件 else 表达式2(python里可以把它们写在一行的用法及numpy.random.choice的用法)

Radinka ·
更新时间:2024-09-20
· 794 次阅读

import numpy as np do = np.random.choice([True,False], p=[0.2,0.8]) do = [0, 1] if do else [1, 0]

今天学习过程中看到了这一行代码,实在太新鲜了,于是就想记录一下(因为从来没见过)
if else 写在一行的。
下面介绍一下这个:
if else 写在一行的格式:
1 if 条件 else 2
规则:条件为真,返回值为1的值,条件为假,返回值为2的值

示例:

import numpy as np do = np.random.choice([True,False], p=[0.2,0.8]) # do是随机数,可能是True,可能是False do = [0, 1] if do else [1, 0]

上面的代码可能返回 [0, 1], 也可能返回[1, 0]

然后就顺便介绍一下 numpy里的random.choice.
这就是一个生成随机数的方法。
参数说明如下:
numpy.random.choice(a, size=None, replace=True, p=None)
a : 是一个一维数组(不能是多维),表示从这个一维数组中随机采样;如果是int型,就表示从0到a-1这个序列中随机采样。也可以是布尔型,也是在一维数组中随机采样。
size : 采样结果的数量,默认为1.可以是整数,表示要采样的数量;也可以为tuple,如(m, n, k),则要采样的数量为m * n * k,size为(m, n, k)。
replace : boolean型,replace指定为True时,采样的元素会有重复;当replace指定为False时,采样不会重复。
p : 一个一维数组,制定了a中每个元素采样的概率,若为默认的None,则a中每个元素被采样的概率相同。

南风无知意 原创文章 20获赞 32访问量 3896 关注 私信 展开阅读全文
作者:南风无知意



NumPy choice IF ELSE random Python

需要 登录 后方可回复, 如果你还没有账号请 注册新账号