python random.choice TypeError: '(array([ 109, 1280, 427, 531, 1563, 102, 1774, 802, 560, 0]), slice(None, None, None))' is an invalid key

Question

所以我正在尝试读取 excel 文档，然后读取 select x 随机且未替换的行数。当我尝试运行时遇到错误，希望得到一些指导。我正在使用 VS Code 编写 Jupyter Notebook。

#import libraries.
import os
import subprocess
import sys
import pandas as pd
import numpy as np
import tkinter as tk

#allow user to browse for specific excel file
from tkinter import filedialog
root = tk.Tk()
root.withdraw()
file_path = filedialog.askopenfilename()
sizeOfSample = 10

#read in excel as dataframe after user selects file in explorer
df = pd.read_excel (file_path)

#select random rows from df to display.
number_of_rows = df.shape[0]
random_indices = np.random.choice(number_of_rows, size=sizeOfSample, replace=False)
random_rows = df[random_indices, :]

print (random_rows)

这是我得到的输出。

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
~\AppData\Local\Temp/ipykernel_1716/1509119795.py in <module>
     21 number_of_rows = initArr.shape[0]
     22 random_indices = np.random.choice(number_of_rows, size=sizeOfSample, replace=False)
---> 23 random_rows = initArr[random_indices, :]
     24 
     25 print (random_rows)

C:\Python39\lib\site-packages\pandas\core\frame.py in __getitem__(self, key)
   3456             if self.columns.nlevels > 1:
   3457                 return self._getitem_multilevel(key)
-> 3458             indexer = self.columns.get_loc(key)
   3459             if is_integer(indexer):
   3460                 indexer = [indexer]

C:\Python39\lib\site-packages\pandas\core\indexes\base.py in get_loc(self, key, method, tolerance)
   3359             casted_key = self._maybe_cast_indexer(key)
   3360             try:
-> 3361                 return self._engine.get_loc(casted_key)
   3362             except KeyError as err:
   3363                 raise KeyError(key) from err

C:\Python39\lib\site-packages\pandas\_libs\index.pyx in pandas._libs.index.IndexEngine.get_loc()

C:\Python39\lib\site-packages\pandas\_libs\index.pyx in pandas._libs.index.IndexEngine.get_loc()

TypeError: '(array([ 109, 1280,  427,  531, 1563,  102, 1774,  802,  560,    0]), slice(None, None, None))' is an invalid key

Answer 1

替换：

random_rows = df[random_indices, :]

作者：

random_rows = df.loc[random_indices, :]

但是你可以使用：

random_rows = df.sample(n=sizeOfSample, replace=True)

python random.choice TypeError: '(array([ 109, 1280, 427, 531, 1563, 102, 1774, 802, 560, 0]), slice(None, None, None))' is an invalid key

python random.choice TypeError: '(array([ 109, 1280, 427, 531, 1563, 102, 1774, 802, 560, 0]), slice(None, None, None))' is an invalid key

numpy

pandas

jupyter-notebook