Как выбрать определенный столбец массива Numpy, считанный из CSV?

Question

Я пытаюсь:

import numpy as np

housing_data = np.loadtxt('Housing.csv', delimiter=',')
print(housing_data)
print(housing_data.shape)
x1 = housing_data[:,:,0]
x2 = housing_data[:,:,1]
y = housing_data[:,:,2]

print(x1)
print(x2)
print(y)

Мои данные имеют форму (47, 3) и выглядят так:

[[2.104e+03 3.000e+00 3.999e+05]
 [1.600e+03 3.000e+00 3.299e+05]
 [2.400e+03 3.000e+00 3.690e+05]
 ....

Я пытаюсь установить для первого столбца значение x1,второй до x2 и третий до y.Но мой код не работает.Что я делаю не так?

Answer 1

Я создал фиктивный * CSV-файл со случайными данными.Я бы сделал это так:

import numpy as np
import pandas as pd

# read file using pandas, without header and convert it to numpy arrays
housing_data = pd.read_csv('Housing.csv', header=None).values

# print housing data
print(housing_data)
print(housing_data.shape)

# slice through the data
x1 = housing_data[:,0]
x2 = housing_data[:,1]
y = housing_data[:,2]

print(x1)
print(x2)
print(y)

Вывод выглядит так:

Answer 2

выбор с Numpy & Python, который вы можете использовать:

#Shape (2,2) from top right corner
data[:2,1:]

#Shape bottom row
data[2]

#Shape bottom row
data[2,:]

или с условиями:

data[data>2]

Возможно, вы можете проверить ваш файл .csv и типы данных:

data.astype(float) 
data = np.arange(3, dtype=np.uint8)