Welcome to OGeek Q&A Community for programmer and developer-Open, Learning and Share
Welcome To Ask or Share your Answers For Others

Categories

0 votes
4.8k views
in Technique[技术] by (71.8m points)

numpy - Divide columns in array without loops

For example:

Input:

a = array([[1, 2], [4, 10], [4, 6]]])

Output:

b = array([[4, 5], [1, 0.6]])

a can have more than three rows.


与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…
Welcome To Ask or Share your Answers For Others

1 Reply

0 votes
by (71.8m points)

You can achieve this using np.roll, by dividing a by a-shifted:

>>> a = np.array([[1,2], [4,10], [4,6]])
array([[ 1,  2],
       [ 4, 10],
       [ 4,  6]])

>>> shifted_a = np.roll(a, shift=1, axis=0)
array([[ 4,  6],
       [ 1,  2],
       [ 4, 10]])

>>> shifted_a / a
array([[0.25      , 0.33333333],
       [4.        , 5.        ],
       [1.        , 0.6       ]])

Then discard the first rows:

>>> (shifted_a / a)[1:]
array([[4. , 5. ],
       [1. , 0.6]])

与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…
OGeek|极客中国-欢迎来到极客的世界,一个免费开放的程序员编程交流平台!开放,进步,分享!让技术改变生活,让极客改变未来! Welcome to OGeek Q&A Community for programmer and developer-Open, Learning and Share
Click Here to Ask a Question

1.4m articles

1.4m replys

5 comments

57.0k users

...