Scikit-learn Bunch 객체의 데이터를 Pandas DataFrame으로 어떻게 변환합니까?
from sklearn.datasets import load_iris
import pandas as pd
data = load_iris()
print(type(data))
data1 = pd. # Is there a Pandas method to accomplish this?
답변:
수동으로 pd.DataFrame
생성자 를 사용 하여 numpy 배열 ( data
)과 열 이름 목록 ( )을 제공 할 수 있습니다 columns
. 당신은 기능과 하나 개의 NumPy와 배열에 대상 연결할 수 있습니다, 하나 DataFrame에 모든 것을 가지고하려면 np.c_[...]
합니다 (주의를 []
)
import numpy as np
import pandas as pd
from sklearn.datasets import load_iris
# save load_iris() sklearn dataset to iris
# if you'd like to check dataset type use: type(load_iris())
# if you'd like to view list of attributes use: dir(load_iris())
iris = load_iris()
# np.c_ is the numpy concatenate function
# which is used to concat iris['data'] and iris['target'] arrays
# for pandas column argument: concat iris['feature_names'] list
# and string list (in this case one string); you can make this anything you'd like..
# the original dataset would probably call this ['Species']
data1 = pd.DataFrame(data= np.c_[iris['data'], iris['target']],
columns= iris['feature_names'] + ['target'])
load_boston()
. 이 답변은 더 일반적으로 작동합니다. stackoverflow.com/a/46379878/1840471
from sklearn.datasets import load_iris
import pandas as pd
data = load_iris()
df = pd.DataFrame(data.data, columns=data.feature_names)
df.head()
이 튜토리얼에 관심이있을 수 있습니다 : http://www.neural.cz/dataset-exploration-boston-house-pricing.html
TOMDLt의 솔루션은 scikit-learn의 모든 데이터 세트에 대해 충분히 일반적이지 않습니다. 예를 들어 보스턴 주택 데이터 세트에서는 작동하지 않습니다. 더 보편적 인 다른 솔루션을 제안합니다. numpy도 사용할 필요가 없습니다.
from sklearn import datasets
import pandas as pd
boston_data = datasets.load_boston()
df_boston = pd.DataFrame(boston_data.data,columns=boston_data.feature_names)
df_boston['target'] = pd.Series(boston_data.target)
df_boston.head()
일반적인 기능으로 :
def sklearn_to_df(sklearn_dataset):
df = pd.DataFrame(sklearn_dataset.data, columns=sklearn_dataset.feature_names)
df['target'] = pd.Series(sklearn_dataset.target)
return df
df_boston = sklearn_to_df(datasets.load_boston())
이것을 알아내는 데 2 시간이 걸렸습니다.
import numpy as np
import pandas as pd
from sklearn.datasets import load_iris
iris = load_iris()
##iris.keys()
df= pd.DataFrame(data= np.c_[iris['data'], iris['target']],
columns= iris['feature_names'] + ['target'])
df['species'] = pd.Categorical.from_codes(iris.target, iris.target_names)
내 판다의 종을 되찾아
그렇지 않으면 실제 pandas 데이터 프레임 인 seaborn 데이터 세트 를 사용 하십시오 .
import seaborn
iris = seaborn.load_dataset("iris")
type(iris)
# <class 'pandas.core.frame.DataFrame'>
scikit learn 데이터 세트와 비교 :
from sklearn import datasets
iris = datasets.load_iris()
type(iris)
# <class 'sklearn.utils.Bunch'>
dir(iris)
# ['DESCR', 'data', 'feature_names', 'filename', 'target', 'target_names']
기능과 대상 변수를 결합하는 다른 방법은 np.column_stack
( 세부 정보 )
import numpy as np
import pandas as pd
from sklearn.datasets import load_iris
data = load_iris()
df = pd.DataFrame(np.column_stack((data.data, data.target)), columns = data.feature_names+['target'])
print(df.head())
결과:
sepal length (cm) sepal width (cm) petal length (cm) petal width (cm) target
0 5.1 3.5 1.4 0.2 0.0
1 4.9 3.0 1.4 0.2 0.0
2 4.7 3.2 1.3 0.2 0.0
3 4.6 3.1 1.5 0.2 0.0
4 5.0 3.6 1.4 0.2 0.0
당신의 캐릭터 라인 라벨을 필요로하는 경우 target
에, 당신은 사용할 수있는 replace
변환 target_names
에 dictionary
새로운 열을 추가 :
df['label'] = df.target.replace(dict(enumerate(data.target_names)))
print(df.head())
결과:
sepal length (cm) sepal width (cm) petal length (cm) petal width (cm) target label
0 5.1 3.5 1.4 0.2 0.0 setosa
1 4.9 3.0 1.4 0.2 0.0 setosa
2 4.7 3.2 1.3 0.2 0.0 setosa
3 4.6 3.1 1.5 0.2 0.0 setosa
4 5.0 3.6 1.4 0.2 0.0 setosa
버전 0.23부터는 as_frame
인수를 사용하여 DataFrame을 직접 반환 할 수 있습니다 . 예를 들어 홍채 데이터 세트를로드합니다.
from sklearn.datasets import load_iris
iris = load_iris(as_frame=True)
df = iris.data
임시 릴리스 노트를 사용하여 이해 하면 breast_cancer, 당뇨병, 손가락, 홍채, linnerud, wine 및 california_houses 데이터 세트에 대해 작동합니다.
매개 변수 as_frame=True
를 사용하여 pandas 데이터 프레임을 가져올 수 있습니다 .
from sklearn import datasets
X,y = datasets.load_iris(return_X_y=True) # numpy arrays
dic_data = datasets.load_iris(as_frame=True)
print(dic_data.keys())
df = dic_data['frame'] # pandas dataframe data + target
df_X = dic_data['data'] # pandas dataframe data only
ser_y = dic_data['target'] # pandas series target only
dic_data['target_names'] # numpy array
from sklearn import datasets
fnames = [ i for i in dir(datasets) if 'load_' in i]
print(fnames)
fname = 'load_boston'
loader = getattr(datasets,fname)()
df = pd.DataFrame(loader['data'],columns= loader['feature_names'])
df['target'] = loader['target']
df.head(2)
베스트 답변을 작성하고 내 의견을 처리하면 여기에 변환 기능이 있습니다.
def bunch_to_dataframe(bunch):
fnames = bunch.feature_names
features = fnames.tolist() if isinstance(fnames, np.ndarray) else fnames
features += ['target']
return pd.DataFrame(data= np.c_[bunch['data'], bunch['target']],
columns=features)
TomDLT가 대답 한 것이 무엇이든간에 일부 사용자에게는 작동하지 않을 수 있습니다.
data1 = pd.DataFrame(data= np.c_[iris['data'], iris['target']],
columns= iris['feature_names'] + ['target'])
iris [ 'feature_names']는 numpy 배열을 반환하기 때문입니다. numpy 배열에서는 + 연산자만으로 배열과 목록 [ 'target']을 추가 할 수 없습니다. 따라서 먼저 목록으로 변환 한 다음 추가해야합니다.
넌 할 수있어
data1 = pd.DataFrame(data= np.c_[iris['data'], iris['target']],
columns= list(iris['feature_names']) + ['target'])
이것은 잘 작동합니다 ..
더 나은 방법이있을 수 있지만 여기에 내가 과거에 한 일이 있으며 꽤 잘 작동합니다.
items = data.items() #Gets all the data from this Bunch - a huge list
mydata = pd.DataFrame(items[1][1]) #Gets the Attributes
mydata[len(mydata.columns)] = items[2][1] #Adds a column for the Target Variable
mydata.columns = items[-1][1] + [items[2][0]] #Gets the column names and updates the dataframe
이제 mydata에는 속성, 대상 변수 및 열 이름 등 필요한 모든 것이 있습니다.
mydata = pd.DataFrame(items[1][1])
슬로우TypeError: 'dict_items' object does not support indexing
이 스 니펫은 TomDLT 및 rolyat 가 이미 기여하고 설명한 것을 기반으로 구축 된 구문 설탕 일뿐 입니다. 유일한 차이점은 load_iris
사전 대신 튜플을 반환하고 열 이름이 열거 된다는 것입니다.
df = pd.DataFrame(np.c_[load_iris(return_X_y=True)])
import pandas as pd
from sklearn.datasets import load_iris
iris = load_iris()
X = iris['data']
y = iris['target']
iris_df = pd.DataFrame(X, columns = iris['feature_names'])
iris_df.head()
나는 귀하의 답변에서 몇 가지 아이디어를 가져 왔으며 더 짧게 만드는 방법을 모르겠습니다. :)
import pandas as pd
from sklearn.datasets import load_iris
iris = load_iris()
df = pd.DataFrame(iris.data, columns=iris['feature_names'])
df['target'] = iris['target']
이렇게하면 feature_names와 대상이 열 및 RangeIndex (start = 0, stop = len (df), step = 1) 인 Pandas DataFrame이 제공됩니다. 'target'을 직접 추가 할 수있는 더 짧은 코드를 갖고 싶습니다.
from sklearn.datasets import load_iris
import pandas as pd
iris_dataset = load_iris()
datasets = pd.DataFrame(iris_dataset['data'], columns =
iris_dataset['feature_names'])
target_val = pd.Series(iris_dataset['target'], name =
'target_values')
species = []
for val in target_val:
if val == 0:
species.append('iris-setosa')
if val == 1:
species.append('iris-versicolor')
if val == 2:
species.append('iris-virginica')
species = pd.Series(species)
datasets['target'] = target_val
datasets['target_name'] = species
datasets.head()
이것은 나를 위해 일한 쉬운 방법입니다.
boston = load_boston()
boston_frame = pd.DataFrame(data=boston.data, columns=boston.feature_names)
boston_frame["target"] = boston.target
그러나 이것은 load_iris에도 적용될 수 있습니다.