Преобразовать столбец с плавающей точкой в ​​int на основе списка столбцов Pandas - PullRequest
Новая
       10

Преобразовать столбец с плавающей точкой в ​​int на основе списка столбцов Pandas

0 голосов
/ 10 января 2020

Какой самый быстрый способ преобразования типов данных столбцов в pandas основан на списке столбцов ниже. Я только хочу преобразовать указанный список в целые числа из чисел с плавающей точкой. Список является Dynami c, используется в функции, поэтому мне нужно, чтобы он был основан на сгенерированном списке

У меня есть код ниже, но я хочу упростить его 

df = pd.DataFrame(data={'a' : [1.0,2.0,3.0],
    'b' : [4.0,5.0,6.0],
    'c' : [7.0,8.0,9.0]})

df

    a   b   c
0   1.0 4.0 7.0
1   2.0 5.0 8.0
2   3.0 6.0 9.0

test_df.dtypes

a    float64
b    float64
c    float64

col_list = ['a', 'b']

col_dtype_zip = zip(col_list, ['int32']*len(col_list))

col_dtype_dict = {key:value for (key,value) in col_dtype_zip}

test_df = test_df.astype(col_dtype_dict)

test_df.dtypes

dtype: object
a      int32
b      int32
c    float64
...