如何在 Pandas 中創建 DataFrame

Question

我正在使用 playerStat.csv，其中包含 8 列，而我只需要其中 2 列。因此我正在嘗試創建一個僅包含這 2 列的新 DataFrame。

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
dataset = pd.read_csv("HLTVData/playerStats.csv")
dataset.head(20)

我只需要 ADR 和評級。

所以我首先用數據集創建一個矩陣。

mat = dataset.as_matrix()
#4 is the ADR and 6 is the Rating
newDAtaSet = pd.DataFrame(dataset, index=indexMatrix,columns=(mat[:,4],mat[:,6]) )

但是沒用，拋出異常

NameError                                 Traceback (most recent call last)
<ipython-input-10-1f975cc2514a> in <module>()
      1 #4 is the ADR and 6 is the Rating
----> 2 newDataSet  = pd.DataFrame(dataset, index=indexMatrix,columns=(mat[:,4],mat[:,6]) )
      

NameError: name 'indexMatrix' is not defined

我也嘗試使用數據集。

newDataSet = pd.DataFrame(dataset, index=np.array(range(dataset.shape[0])), columns=dataset['ADR'])

/home/tensor/miniconda3/envs/tensorflow35openvc/lib/python3.5/site-packages/pandas/core/internals.py in _make_na_block(self, placement, fill_value)
   3984 
   3985         dtype, fill_value = infer_dtype_from_scalar(fill_value)
-> 3986         block_values = np.empty(block_shape, dtype=dtype)
   3987         block_values.fill(fill_value)
   3988         return make_block(block_values, placement=placement)

MemoryError: 

Answer 1

我認為您需要在read_csv usecols

dataset = pd.read_csv("HLTVData/playerStats.csv", usecols=['ADR','Rating'])

要么：

dataset = pd.read_csv("HLTVData/playerStats.csv", usecols=[4,6])