FilesExpand file tree

 master

/

numpy_bool_index.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

31 lines (21 loc) · 605 Bytes

 master

/

numpy_bool_index.py

Copy path

Top

File metadata and controls

• Code
• Blame

31 lines (21 loc) · 605 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

import numpy as np

if __name__ == "__main__":

names = np.array(["Bob", "Joe", "Will", "Bob", "Will", "Joe", "Joe"])

data = np.random.randn(7, 4)

print(names)

print(data)

print(names == "Bob")

print(data[names == "Bob"])

print(data[names == "Bob", 2:])

print(data[names == "Bob", 3])

print(names != "Bob")

print(data[~(names == "Bob")])

cond = names == "Bob"

print(data[~cond])

mask = (names == "Bob") | (names == "Will")

print(mask)

print(data[mask])

data[data < 0] = 0

print(data)

data[names != "Joe"] = 7

print(data)