Python：在數據幀中將字符串數組轉換為int數組

Question

我有一個數據框，持續時間是其中一個屬性。 持續時間的內容如下：

            array(['487', '346', ...,  '227', '17']). 

而df.info（），我得到：數據列（共22列）：

             duration        2999 non-null object
             campaign        2999 non-null object
             ...

現在我想將持續時間轉換為int。 有什么解決方案嗎？

Answer 1

使用astype ：

df['duration'] = df['duration'].astype(int)

計時

使用以下設置生成大型樣本數據集：

n = 10**5
data = list(map(str, np.random.randint(10**4, size=n)))
df = pd.DataFrame({'duration': data})

我得到以下時間：

%timeit -n 100 df['duration'].astype(int)
100 loops, best of 3: 10.9 ms per loop

%timeit -n 100 df['duration'].apply(int)
100 loops, best of 3: 44.3 ms per loop

%timeit -n 100 df['duration'].apply(lambda x: int(x))
100 loops, best of 3: 60.1 ms per loop

Answer 2

df['duration'] = df['duration'].astype(int)

Answer 3

使用int(str) ：

df['duration'] = df['duration'].apply(lambda x: int(x)) #df is your dataframe with attribute 'duration'