Python Pandas 替换字符串的一部分

Question

我正在尝试过滤存储在包含时间和角度值的 .csv 文件中的数据，并将过滤后的数据保存在输出 .csv 文件中。我解决了过滤部分，但问题是时间以 hh:mm:ss:msmsmsms (12:55:34:500) 格式记录，我想将其更改为 hhmmss (125534) 或换句话说删除：和毫秒部分。 我尝试使用 .replace 函数，但我不断收到 KeyError: 'time' 错误。

输入数据：

time,angle
12:45:55,56
12:45:56,89
12:45:57,112
12:45:58,189
12:45:59,122
12:46:00,123

代码：

import pandas as pd

#define min and max angle values
alpha_min = 110
alpha_max = 125

#read input .csv file
data = pd.read_csv('test_csv3.csv', index_col=0)

#filter by angle size
data = data[(data['angle'] < alpha_max) & (data['angle'] > alpha_min)]

#replace ":" with "" in time values
data['time'] = data['time'].replace(':','')

#display results
print data

#write results
data.to_csv('test_csv3_output.csv')

Answer 1

那是因为time 是一个索引。您可以这样做并删除 index_col=0:

data = pd.read_csv('test_csv3.csv')

并改变这一行：

data['time'] = pd.to_datetime(data['time']).dt.strftime('%H%M%S')

输出：

     time  angle
2  124557    112
4  124559    122
5  124600    123

Answer 2

print (data.keys()) 或 print(data.head()) 会产生什么？似乎您在 time 索引字符串之前\之后有一个杂散字符，有时会发生，具体取决于 csv 的创建方式与读取方式（请参阅this question）。

如果它不是一个更大的项目和/或您只是想要数据，您可以做一些愚蠢的解决方法，例如：timeKeyString=list(data.columns.values)[0]（假设时间是第一个）。