list comprehension

Using a list comprehension: [x for x in iterable] to create list instance

[expresion for x in iterabel ] 列表推導式的核心在expression.

 1 #!/usr/bin/env python3
 2 #-*- coding:utf-8 -*-
 3 ############################
 4 #File Name: test_list_comprehension.py
 5 #Author: frank
 6 #Mail: [email protected]
 

 7 #Created Time:2018-05-29 20:33:56
 8 ############################
 9 import numpy as np
10 
11 seed = 123
12 
13 #模擬數據集標簽
14 #基於seed產生隨機數
15 rdg = np.random.RandomState(seed)
16 #先生成一個5行2列的隨機數矩陣
17 X = rdg.rand(5,2)
18 #每次取出一行，判斷如果兩個數的和小於1，給a賦值1；如果兩個數的和不小於1,給a賦值
19 #[expression for x in iterable]                                                
 

20 Y = [[int(x0+x1<1)] for (x0,x1) in X]
21 print("X:{}".format(X))
22 print("Y:{}".format(Y))

列表推推導式有很強的擴展性, expression 不同，可執行的功能就大不一樣。

python-list comprehension