KeyError: 0 Pandas

Question

Ich brauche Hilfe. Ich versuche, herauszufiltern und in eine andere CSV-Datei zu schreiben, die aus Daten besteht, die nach 10769s in der Spalte elapsed_seconds zusammen mit der Beschleunigungsgröße gesammelt wurden. Aber ich bin immer KeyError: 0 ...

import numpy as np 
import pandas as pd 
import matplotlib.pyplot as plt 

data = pd.read_csv(accelDataPath) 
data.columns = ['t', 'x', 'y', 'z'] 

# calculate the magnitude of acceleration 
data['m'] = np.sqrt(data['x']**2 + data['y']**2 + data['z']**2) 

data['datetime'] = pd.DatetimeIndex(pd.to_datetime(data['t'], unit = 'ms').dt.tz_localize('UTC').dt.tz_convert('US/Eastern')) 
data['elapsed_seconds'] = (data['datetime'] - data['datetime'].iloc[0]).dt.total_seconds() 
i=0 
csv = open("filteredData.csv", "w+") 
csv.write("Event at, Magnitude \n") 
while (i < len(data[data.elapsed_seconds > 10769])): 
    csv.write(str(data[data.elapsed_seconds > 10769][i]) + ", " + str(data[data.m][i]) + "\n") 
csv.close() 

Fehler, die ich erhalte ist:

Traceback (most recent call last): 
    File "C:\Users\Desktop\AnalyzingData.py", line 37, in <module> 
csv.write(str(data[data.elapsed_seconds > 10769][i]) + ", " + str(data[data.m][i]) + "\n") 
    File "C:\python\lib\site-packages\pandas\core\frame.py", line 1964, in __getitem__ 
    return self._getitem_column(key) 
    File "C:\python\lib\site-packages\pandas\core\frame.py", line 1971, in _getitem_column 
    return self._get_item_cache(key) 
    File "C:\python\lib\site-packages\pandas\core\generic.py", line 1645, in _get_item_cache 
    values = self._data.get(item) 
    File "C:\python\lib\site-packages\pandas\core\internals.py", line 3590, in get 
    loc = self.items.get_loc(item) 
    File "C:\python\lib\site-packages\pandas\core\indexes\base.py", line 2444, in get_loc 
    return self._engine.get_loc(self._maybe_cast_indexer(key)) 
    File "pandas\_libs\index.pyx", line 132, in pandas._libs.index.IndexEngine.get_loc (pandas\_libs\index.c:5280) 
     File "pandas\_libs\index.pyx", line 154, in pandas._libs.index.IndexEngine.get_loc (pandas\_libs\index.c:5126) 
     File "pandas\_libs\hashtable_class_helper.pxi", line 1210, in pandas._libs.hashtable.PyObjectHashTable.get_item (pandas\_libs\hashtable.c:20523) 
     File "pandas\_libs\hashtable_class_helper.pxi", line 1218, in pandas._libs.hashtable.PyObjectHashTable.get_item (pandas\_libs\hashtable.c:20477) 
    KeyError: 0 

Answer 1

Änderung dieser Linie

csv.write(
    str(data[data.elapsed_seconds > 10769][i]) + ", " + str(data[data.m][i]) + "\n" 
    ) 

Um dies:

csv.write(
    str(data[data.elapsed_seconds > 10769].iloc[i]) + ", " + str(data[data.m].iloc[i]) +"\n" 
    ) 

Beachten Sie auch, dass Sie nichterhöhen 210, wie diese i += 1, in der While-Schleife.

---

Oder besser, verwenden Sie df.to_csv wie folgt:

import numpy as np 
import pandas as pd 
import matplotlib.pyplot as plt 

data = pd.read_csv(accelDataPath) 
data.columns = ['t', 'x', 'y', 'z'] 

# calculate the magnitude of acceleration 
data['m'] = np.sqrt(data['x']**2 + data['y']**2 + data['z']**2) 

data['datetime'] = pd.DatetimeIndex(pd.to_datetime(data['t'], unit = 'ms').dt.tz_localize('UTC').dt.tz_convert('US/Eastern')) 
data['elapsed_seconds'] = (data['datetime'] - data['datetime'].iloc[0]).dt.total_seconds() 

# write to csv using data.to_csv 
data[data.elapsed_seconds > 10769][['elapsed_seconds', 'm']].to_csv("filteredData.csv", 
      sep=",", 
      index=False)