Extracting specific columns in numpy array

前端 未结 9 1290
隐瞒了意图╮
隐瞒了意图╮ 2020-11-28 02:27

This is an easy question but say I have an MxN matrix. All I want to do is extract specific columns and store them in another numpy array but I get invalid syntax errors. He

9条回答
  •  野性不改
    2020-11-28 02:34

    I assume you wanted columns 1 and 9?

    To select multiple columns at once, use

    X = data[:, [1, 9]]
    

    To select one at a time, use

    x, y = data[:, 1], data[:, 9]
    

    With names:

    data[:, ['Column Name1','Column Name2']]
    

    You can get the names from data.dtype.names

提交回复
热议问题