Pandas SQL操作

Pandas SQL操作的具體實例

由于許多潛在的Pandas用戶都對SQL有所了解，因此本頁面旨在提供一些示例說明如何使用Pandas執(zhí)行各種SQL操作。

 import pandas as pd
 url = 'https://raw.github.com/pandasdev/
 pandas/master/pandas/tests/data/tips.csv'
 tips=pd.read_csv(url)
 print tips.head()

運行結(jié)果如下：

      total_bill   tip      sex  smoker  day     time  size
0        16.99  1.01   Female      No  Sun  Dinner      2
1        10.34  1.66     Male      No  Sun  Dinner      3
2        21.01  3.50     Male      No  Sun  Dinner      3
3        23.68  3.31     Male      No  Sun  Dinner      2
4        24.59  3.61   Female      No  Sun  Dinner      4

查詢

在SQL中，選擇是使用您選擇的列的逗號分隔列表（或使用*來選擇所有列）來完成的：

 SELECT total_bill, tip, smoker, time
 from tips
 LIMIT 5;

使用Pandas，通過將列名稱列表傳遞到DataFrame來完成列選擇：

 tips[['total_bill', 'tip', 'smoker', 'time']].head(5)

讓我們看一個完整的實例：

 import pandas as pd
 url = 'https://raw.github.com/pandasdev/
 pandas/master/pandas/tests/data/tips.csv'
  
 tips=pd.read_csv(url)
 print tips[['total_bill', 'tip', 'smoker', 'time']].head(5)

運行結(jié)果如下：

   total_bill   tip  smoker     time
0       16.99  1.01      No   Dinner
1       10.34  1.66      No   Dinner
2       21.01  3.50      No   Dinner
3       23.68  3.31      No   Dinner
4       24.59  3.61      No   Dinner

調(diào)用不帶列名列表的DataFrame將顯示所有列（類似于SQL的*）。

WHERE條件查詢

通過WHERE子句在SQL中進行過濾。

 SELECT * from tips WHERE time = 'Dinner' LIMIT 5;

DataFrame可以通過多種方式進行過濾。最直觀的方法是使用布爾索引。

 tips[tips['time'] == 'Dinner'].head(5)

我們來看一個完整的實例

 import pandas as pd
 url = 'https://raw.github.com/pandasdev/
 pandas/master/pandas/tests/data/tips.csv'
 tips=pd.read_csv(url)
 print tips[tips['time'] == 'Dinner'].head(5)

運行結(jié)果如下：

     total_bill   tip      sex  smoker  day    time  size
0       16.99  1.01   Female     No   Sun  Dinner    2
1       10.34  1.66     Male     No   Sun  Dinner    3
2       21.01  3.50     Male     No   Sun  Dinner    3
3       23.68  3.31     Male     No   Sun  Dinner    2
4       24.59  3.61   Female     No   Sun  Dinner    4

上面的語句將一系列True / False對象傳遞給DataFrame，并返回所有帶有True的行。

GroupBy分組

此操作獲取整個數(shù)據(jù)集中每個組中的記錄數(shù)。例如查詢性別分組和數(shù)量：

 SELECT sex, count(*)
 from tips
 GROUP BY sex;

在Pandas是如下操作：

 tips.groupby('sex').size()

我們來看一個完整的實例

 import pandas as pd
 url = 'https://raw.github.com/pandasdev/
 pandas/master/pandas/tests/data/tips.csv'
 tips=pd.read_csv(url)
 print tips.groupby('sex').size()

運行結(jié)果如下：

 sex
 Female 87
 Male 157
 dtype: int64

查詢N行數(shù)量

SQL 使用LIMIT返回N行：

 SELECT * from tips
 LIMIT 5 ;

在Pandas中操作如下：

 tips.head(5)

我們來看一個完整的實例

 import pandas as pd
 url = 'https://raw.github.com/pandas-dev/pandas/master/pandas/tests/data/tips.csv'
 tips=pd.read_csv(url)
 tips = tips[['smoker', 'day', 'time']].head(5)
 print tips

運行結(jié)果如下：

     smoker   day     time
0      No   Sun   Dinner
1      No   Sun   Dinner
2      No   Sun   Dinner
3      No   Sun   Dinner
4      No   Sun   Dinner