IT评测·应用市场-qidao123.com

标题: 【pandas基础】--数据类型 [打印本页]

作者: 盛世宏图    时间: 2023-5-27 13:48
标题: 【pandas基础】--数据类型
数据类型是计算机编程中将不同类型的数据值分类和定义的方式。
通过数据类型,可以确定数据的存储方式和内存占用量,了解不同类型的数据进行各种运算的能力。
使用pandas进行数据分析时,最常用到的几种类型是:
1. 字符串类型

pandas字符串类型主要用于处理文本数据或包含文本数据的列。
它可以快速、方便地对文本数据进行操作,比如:
字符串在python中是str类型。
  1. In [1]: s = "hello"
  2. In [2]: type(s)
  3. Out[2]: str
复制代码
但是在pandas的DataFrame中则是object类型。
  1. import pandas as pd
  2. df = pd.DataFrame(
  3.     {
  4.         "name": ["小华", "小红", "小明"],
  5.         "age": [12, 15, 13],
  6.         "score": [80.5, 98.5, 80],
  7.     },
  8. )
  9. df.dtypes
复制代码

为什么在pandas中,字符串是object类型呢?
因为pandas中的数据类型继承自numpy的ndarray,ndarray的每个元素都必须明确占用内存的大小。
对于int64和float64来说,它们都占用8个字节的内存,而字符串由于长度不固定,无法确定占用内存的大小,所以都用object类型,这个object类型可以看做是一个指向实际存储字符串位置的的指针。
2. 数值类型

数值类型有两种,一种是整数,一种是浮点数(也就是平时说的小数)。
一般来说,各类分析算法以及可视化展示需要的都是数值类型,数值类型是我们分析数据时使用最多的部分。
上面的示例中,age和score列分别是整数和浮点数类型。
  1. import pandas as pd
  2. df = pd.DataFrame(
  3.     {
  4.         "name": ["小华", "小红", "小明"],
  5.         "age": [12, 15, 13],
  6.         "score": [80.5, 98.5, 80],
  7.     },
  8. )
  9. df.dtypes
复制代码

DataFrame中数值类型默认是64位的,可以存储更大的数字。
3. catagory 类型

pandas中的category类型是一种用于处理分类变量的数据类型。
它可以大大提高数据处理和计算效率,并减少内存占用。
在某些情况下,数据中的一些变量只包含有限的可能取值,例如“性别”、“地区”等,这些变量可以归类为分类变量。
如果将这些变量存储为字符串或数字形式,则可能会浪费大量的内存,因为每个变量都会占据大量的空间。
这就是category类型的用处:使用category类型可以将这些变量存储为原始数据的唯一值的散列表,从而大大减少了内存占用。
除了内存优化外,category类型还提供了一些便捷的方法来处理分类变量,例如自动排序和类别之间的比较。
因此,如果数据中包含分类变量,则应该使用category类型来优化数据处理和计算效率。
下面的示例,使用中国人口统计的相关数据,默认导入之后数据情况如下:
  1. import pandas as pd
  2. fp = "http://databook.top:8888/pandas/cn-people.csv"
  3. df = pd.read_csv(fp)
  4. df
复制代码

各个列的默认类型如下:
  1. df.dtypes
复制代码

其中【指标编码】和【指标中文】列的类型其实是字符串。
各个列实际占用的内存大小:
  1. df.memory_usage(deep=True)
复制代码

Index表示索引所占用的内存大小,可以看出【指标编码】和【指标中文】占用的内存比较多,而且这两列重复数据也比较多。
尝试将【指标编码】和【指标中文】两列转换为catagory类型之后,看看内存占用是否减少。
  1. df["指标中文"] = df["指标中文"].astype("category")
  2. df["指标编码"] = df["指标编码"].astype("category")
  3. df.dtypes
复制代码

类型已经转换成功,看看pandas是如何给catagory类型编码的。
  1. df["指标中文"].values.codes
复制代码

可以看出,是用int8类型来编码,int8类型只占用1个字节的内存,总体应该能够节省不少内存空间。
  1. df.memory_usage(deep=True)
复制代码

【指标编码】和【指标中文】两列的内存占用只有原来的约1/200。
4. 类型间的转换

pandas中的类型转换操作可以将一种数据类型转换为另一种数据类型,以便更好地处理和分析数据。
数据类型的选择会影响数据的存储方式和计算速度,因此,在不同的情况下,数据类型的选择是非常重要的,正确的选择可以有效地提高代码的性能和准确性。
例如,将文本数据转换为数字数据,可以使得数据更容易进行数值运算和可视化,从而方便地做出相关的决策和分析。
类型转换常用的两种方式是astype函数和自定义函数。
4.1 astype

类型转换最常用的方法是astype,前面介绍catagory类型时,示例中已经演示了字符串类型到catagory类型的转换。
数值类型之间,或者数值类型和字符串类型之间也是可以互相转换的。
  1. df = pd.DataFrame(
  2.     {
  3.         "name": ["小华", "小红", "小明"],
  4.         "age": [12, 15, 13],
  5.         "score": [80.5, 98.5, 80],
  6.     },
  7. )
  8. print(df.dtypes)
  9. # int64 ==> float64
  10. df.age = df.age.astype("float64")
  11. # float64 ==> string
  12. df.score = df.score.astype("str")
  13. print(df.dtypes)
复制代码

4.2 自定义函数

字符串类型也是可以转换成数值类型的,前提是字符串的内容得是数值。
  1. df = pd.DataFrame(
  2.     {
  3.         "name": ["小华", "小红", "小明"],
  4.         "money": ["¥55", "¥12", "¥58"],
  5.         "score": ["80.5", "98.5", "80"],
  6.     },
  7. )
  8. print(df.dtypes)
  9. # df.name.astype("int64")
  10. # df.money.astype("int64")
  11. df.score = df.score.astype("float64")
  12. print(df.dtypes)
复制代码

score列有字符串类型转换为了float64类型,另外代码中注释的两行是不能成功转换的,去掉注释后,代码执行时会抛出异常错误。
看上面的示例数据,name列是不太可能转成数值类型的,但是money列只是多了一个人民币符号¥,其实这列本质上应该是数值类型,也许后续需要根据这列的数值来分析花费的费用等情况。
这时,直接用astype是无法完成类型转换的,要用自定义函数来去掉人民币符号¥,再转换成数值类型。
  1. df = pd.DataFrame(
  2.     {
  3.         "name": ["小华", "小红", "小明"],
  4.         "money": ["¥55", "¥12", "¥58"],
  5.         "score": ["80.5", "98.5", "80"],
  6.     },
  7. )
  8. convert = lambda s: float(s.replace("¥", ""))
  9. print(df.dtypes)
  10. df.money = df.money.apply(convert)
  11. print(df.dtypes)
复制代码

通过自定义的convert函数,money列成功转换成了float64类型。
5. 总结回顾

这里只是介绍了最常见的数据类型,还有日期类型也很重要,下一篇单独介绍。
除此之外,细分的话,还有以下各类数据类型:

免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!




欢迎光临 IT评测·应用市场-qidao123.com (https://dis.qidao123.com/) Powered by Discuz! X3.4