12个Numpy与Pandas高效技巧,让数据分析更容易

开发 后端
本文分享给大家 12 种 Numpy 和 Pandas 函数,这些高效的函数会令数据分析更为容易、便捷。最后,读者也可以在 GitHub 项目中找到本文所用代码的 Jupyter Notebook。

 本文分享给大家 12 种 Numpy 和 Pandas 函数,这些高效的函数会令数据分析更为容易、便捷。最后,读者也可以在 GitHub 项目中找到本文所用代码的 Jupyter Notebook。

项目地址:https://github.com/kunaldhariwal/12-Amazing-Pandas-NumPy-Functions

Numpy 的 6 种高效函数

首先从 Numpy 开始。Numpy 是用于科学计算的 Python 语言扩展包,通常包含强大的 N 维数组对象、复杂函数、用于整合 C/C++和 Fortran 代码的工具以及有用的线性代数、傅里叶变换和随机数生成能力。

除了上面这些明显的用途,Numpy 还可以用作通用数据的高效多维容器(container),定义任何数据类型。这使得 Numpy 能够实现自身与各种数据库的无缝、快速集成。

接下来一一解析 6 种 Numpy 函数。

1、argpartition()

借助于 argpartition(),Numpy 可以找出 N 个最大数值的索引,也会将找到的这些索引输出。然后我们根据需要对数值进行排序。 

  1. x = np.array([12, 10, 12, 0, 6, 8, 9, 1, 16, 4, 6, 0])index_val = np.argpartition(x, -4)[-4:]  
  2. index_val  
  3. array([1, 8, 2, 0], dtype=int64)np.sort(x[index_val])  
  4. array([10, 12, 12, 16]) 

2、allclose()

allclose() 用于匹配两个数组,并得到布尔值表示的输出。如果在一个公差范围内(within a tolerance)两个数组不等同,则 allclose() 返回 False。该函数对于检查两个数组是否相似非常有用。 

  1. array1 = np.array([0.12,0.17,0.24,0.29])  
  2. array2 = np.array([0.13,0.19,0.26,0.31])# with a tolerance of 0.1, it should return False:  
  3. np.allclose(array1,array2,0.1)  
  4. False# with a tolerance of 0.2, it should return True: 
  5. np.allclose(array1,array2,0.2)  
  6. True 

3、clip()

Clip() 使得一个数组中的数值保持在一个区间内。有时,我们需要保证数值在上下限范围内。为此,我们可以借助 Numpy 的 clip() 函数实现该目的。给定一个区间,则区间外的数值被剪切至区间上下限(interval edge)。 

  1. x = np.array([3, 17, 14, 23, 2, 2, 6, 8, 1, 2, 16, 0])np.clip(x,2,5)  
  2. array([3, 5, 5, 5, 2, 2, 5, 5, 2, 2, 5, 2]) 

4、extract()

顾名思义,extract() 是在特定条件下从一个数组中提取特定元素。借助于 extract(),我们还可以使用 and 和 or 等条件。 

  1. # Random integers  
  2. array = np.random.randint(20, size=12 
  3. array  
  4. array([ 0,  1,  8, 19, 16, 18, 10, 11,  2, 13, 14,  3])#  Divide by 2 and check if remainder is 1  
  5. cond = np.mod(array, 2)==1  
  6. cond  
  7. array([False,  True, False,  True, False, False, False,  True, False, True, False,  True])# Use extract to get the values  
  8. np.extract(cond, array)  
  9. array([ 1, 19, 11, 13,  3])# Apply condition on extract directly  
  10. np.extract(((array < 3) | (array > 15)), array)  
  11. array([ 0,  1, 19, 16, 18,  2]) 

5、where()

Where() 用于从一个数组中返回满足特定条件的元素。比如,它会返回满足特定条件的数值的索引位置。Where() 与 SQL 中使用的 where condition 类似,如以下示例所示: 

  1. y = np.array([1,5,6,8,1,7,3,6,9])# Where y is greater than 5, returns index position  
  2. np.where(y>5)  
  3. array([2, 3, 5, 7, 8], dtype=int64),)# First will replace the values that match the condition,   
  4. # second will replace the values that does not  
  5. np.where(y>5, "Hit", "Miss")  
  6. array([ Miss ,  Miss ,  Hit ,  Hit ,  Miss ,  Hit ,  Miss ,  Hit ,  Hit ],dtype<U4 ) 

6、percentile()

Percentile() 用于计算特定轴方向上数组元素的第 n 个百分位数。 

  1. a = np.array([1,5,6,8,1,7,3,6,9])print("50th Percentile of a, axis = 0 : ",    
  2.       np.percentile(a, 50, axis =0))  
  3. 50th Percentile of a, axis = 0 :  6.0b = np.array([[10, 7, 4], [3, 2, 1]])print("30th Percentile of b, axis = 0 : ",    
  4.       np.percentile(b, 30, axis =0))  
  5. 30th Percentile of b, axis = 0 :  [5.1 3.5 1.9] 

这就是 Numpy 扩展包的 6 种高效函数,相信会为你带来帮助。接下来看一看 Pandas 数据分析库的 6 种函数。

Pandas的 6 种高效函数

Pandas 也是一个 Python 包,它提供了快速、灵活以及具有显著表达能力的数据结构,旨在使处理结构化 (表格化、多维、异构) 和时间序列数据变得既简单又直观。

Pandas 适用于以下各类数据:

  •  具有异构类型列的表格数据,如 SQL 表或 Excel 表;
  •  有序和无序 (不一定是固定频率) 的时间序列数据;
  •  带有行/列标签的任意矩阵数据(同构类型或者是异构类型);
  •  其他任意形式的统计数据集。事实上,数据根本不需要标记就可以放入 Pandas 结构中。

1、read_csv(nrows=n)

大多数人都会犯的一个错误是,在不需要.csv 文件的情况下仍会完整地读取它。如果一个未知的.csv 文件有 10GB,那么读取整个.csv 文件将会非常不明智,不仅要占用大量内存,还会花很多时间。我们需要做的只是从.csv 文件中导入几行,之后根据需要继续导入。 

  1. import io  
  2. import requests# I am using this online data set just to make things easier for you guys  
  3. url = "https://raw.github.com/vincentarelbundock/Rdatasets/master/csv/datasets/AirPassengers.csv"  
  4. s = requests.get(url).content# read only first 10 rows  
  5. df = pd.read_csv(io.StringIO(s.decode( utf-8 )),nrows=10 , index_col=0

2、map()

map( ) 函数根据相应的输入来映射 Series 的值。用于将一个 Series 中的每个值替换为另一个值,该值可能来自一个函数、也可能来自于一个 dict 或 Series。 

  1. # create a dataframe  
  2. dframe = pd.DataFrame(np.random.randn(4, 3), columns=list( bde ), index=[ India ,  USA ,  China ,  Russia ])#compute a formatted string from each floating point value in frame 
  3. changefn = lambda x:  %.2f  % x# Make changes element-wise 
  4. dframe[ d ].map(changefn) 

3、apply()

apply() 允许用户传递函数,并将其应用于 Pandas 序列中的每个值。 

  1. # max minus mix lambda fn  
  2. fn = lambda x: x.max() - x.min()# Apply this on dframe that we ve just created above  
  3. dframe.apply(fn) 

4、isin()

lsin () 用于过滤数据帧。Isin () 有助于选择特定列中具有特定(或多个)值的行。 

  1. # Using the dataframe we created for read_csv  
  2. filter1 = df["value"].isin([112])   
  3. filter2 = df["time"].isin([1949.000000])df [filter1 & filter2] 

5、copy()

Copy () 函数用于复制 Pandas 对象。当一个数据帧分配给另一个数据帧时,如果对其中一个数据帧进行更改,另一个数据帧的值也将发生更改。为了防止这类问题,可以使用 copy () 函数。 

  1. # creating sample series   
  2. data = pd.Series([ India ,  Pakistan ,  China ,  Mongolia ])# Assigning issue that we face  
  3. datadata1= data  
  4. # Change a value  
  5. data1[0]= USA   
  6. # Also changes value in old dataframe  
  7. data# To prevent that, we use  
  8. # creating copy of series   
  9. new = data.copy()# assigning new values   
  10. new[1]= Changed value # printing data   
  11. print(new)   
  12. print(data) 

6、select_dtypes()

select_dtypes() 的作用是,基于 dtypes 的列返回数据帧列的一个子集。这个函数的参数可设置为包含所有拥有特定数据类型的列,亦或者设置为排除具有特定数据类型的列。 

  1. # We ll use the same dataframe that we used for read_csv  
  2. framex =  df.select_dtypes(include="float64")# Returns only time column 

最后,pivot_table( ) 也是 Pandas 中一个非常有用的函数。如果对 pivot_table( ) 在 excel 中的使用有所了解,那么就非常容易上手了。 

  1. # Create a sample dataframe  
  2. school = pd.DataFrame({ A : [ Jay ,  Usher ,  Nicky ,  Romero ,  Will ],   
  3.        B : [ Masters ,  Graduate ,  Graduate ,  Masters ,  Graduate ],   
  4.        C : [26, 22, 20, 23, 24]})# Lets create a pivot table to segregate students based on age and course  
  5. table = pd.pivot_table(school, values = A , index =[ B ,  C ],   
  6.                          columns =[ B ], aggfunc = np.sum, fill_value="Not Available")   
  7. table  

 

责任编辑:庞桂玉 来源: Linux公社
相关推荐

2020-04-03 13:50:19

数据分析PandasNumPy

2021-12-24 10:45:19

PandasLambda数据分析

2018-08-23 17:15:10

编程语言Python数据分析

2021-02-19 10:59:29

NumpyPandasPython

2009-10-27 09:09:06

Eclipse技巧

2022-09-20 10:50:34

PandasNumPy

2024-01-03 14:54:56

PythonPandas数据处理工具

2022-07-08 06:01:37

D-Tale辅助工具

2023-10-04 00:17:00

SQL数据库

2016-11-18 15:42:06

存储

2020-03-10 08:55:50

PandasNumPy函数

2017-12-27 11:38:14

数据分析大数据算法

2019-09-10 11:31:16

Python数据分析表达式

2020-03-19 15:11:14

Pandas数据分析代码

2022-04-01 06:18:48

数据分析IT领导者

2019-04-29 08:31:25

PythonPandas数据

2019-09-11 14:34:13

排序算法数据科学

2013-01-06 11:01:59

大数据分析

2021-04-15 11:37:47

NumpyPython代码

2023-11-21 09:11:31

点赞
收藏

51CTO技术栈公众号