溫馨提示×

usecols和dtype參數(shù)如何配合

小樊
83
2024-06-14 11:34:32
欄目: 編程語言

usecols參數(shù)用于指定要讀取的列，dtype參數(shù)用于指定每列的數(shù)據(jù)類型。這兩個(gè)參數(shù)可以配合使用，以確保讀取的數(shù)據(jù)類型與指定的數(shù)據(jù)類型一致。

例如，假設(shè)我們有一個(gè)CSV文件，包含兩列數(shù)據(jù)，第一列是整數(shù)，第二列是字符串。我們想要只讀取第一列，并將其數(shù)據(jù)類型指定為浮點(diǎn)型。我們可以這樣使用usecols和dtype參數(shù)：

import pandas as pd

# 讀取CSV文件，只讀取第一列，并將數(shù)據(jù)類型指定為浮點(diǎn)型
df = pd.read_csv('data.csv', usecols=[0], dtype={'column1': float})

print(df)

這樣，我們就只讀取了第一列，并將其數(shù)據(jù)類型指定為浮點(diǎn)型。

0 踩