如何在python

Question

創建數據幀我有表像下面 （在這個例子中，它代表的名單，但這不需要是給出列表）

age=[3,4,5,6] 

sex=[1,2] 

limit=[3,4] 

我想創建數據幀，其有各種組合。

但「限制」列超過「年齡」列。

所以我期望的結果是這樣的。

age sex limit 
3 1 3 
3 1 4 
4 1 4 
3 2 3 
3 2 4 
4 2 4 

我該如何得到這個結果？

Answer 1

此解決方案使用Numpy和Pandas使用'limit'大於或等於'age'的約束創建組合的數據框。請注意，您的帖子中寫道：「大於」，但你的輸出的例子顯示了「大於或等於」：

import itertools 
age=[3,4,5,6] 
sex=[1,2] 
limit=[3,4] 
a = [age, sex, limit] 
mylist = list(itertools.product(*a)) 

import pandas as pd 
import numpy as np 
df = pd.DataFrame(np.array(mylist).reshape(len(mylist),3), columns = ["age","sex","limit"]) 

df2 = df[df['limit'] >= df['age']] 
print df2.to_string(index=False) 

輸出是：

age sex limit 
    3 1  3 
    3 1  4 
    3 2  3 
    3 2  4 
    4 1  4 
    4 2  4   

Answer 2

基本數據：

age=[3,4,5,6] 
sex=[1,2] 
limit=[3,4] 

使用列表理解結合for和if的下列代碼檢查每個組合：

lst = [(a, s, l) 
     for a in age 
     for l in limit 
     if l >= a 
     for s in sex] 
df = pd.DataFrame(data=lst, columns=["age","sex","limit"]) 
print(df) 

輸出：

age sex limit 
0 3 1  3 
1 3 1  4 
2 3 2  3 
3 3 2  4 
4 4 1  4 
5 4 2  4