我正在嘗試將此條件子句放入我的列表比較“arr”中。arr = [[(i*n+j) for i in range(n)] for j in range(n)]和if ( 2*i<=j+i+1<=n+1 and i>0)作為條件。我試著把這個：- 在前面的三元組中有一個 else 語句： "" ，但這會在數(shù)組中產(chǎn)生不需要的元素。-在我的 for 語句之后（我，j，甚至都嘗試過）關(guān)于如何在不增加太多復(fù)雜性的情況下進(jìn)行計算的任何想法？所需的示例輸出：from pandas import DataFrame as dfn = 5arr= [NEEDS HELP HERE]print(df(arr))>>> 0 1 2 3 40 NaN 5 NaN NaN NaN1 NaN 6 11.0 NaN NaN2 NaN 7 12.0 17.0 NaN3 NaN 8 13.0 NaN NaN4 NaN 9 NaN NaN NaN對于 n = 3>>> 0 1 2 0 NaN 3 NaN1 NaN 4 7.02 NaN 5 NaN對于 n = 2>>> 0 1 0 NaN 21 NaN 3對于 n = 10（我的代碼可以生成以下內(nèi)容）>>> 0 1 2 3 4 5 6 7 8 90 _ 10 _ _ _ _ _ _ _ _1 _ 11 21 _ _ _ _ _ _ _2 _ 12 22 32 _ _ _ _ _ _3 _ 13 23 33 43 _ _ _ _ _4 _ 14 24 34 44 54 _ _ _ _5 _ 15 25 35 45 55 _ _ _ _6 _ 16 26 36 46 _ _ _ _ _7 _ 17 27 37 _ _ _ _ _ _8 _ 18 28 _ _ _ _ _ _ _9 _ 19 _ _ _ _ _ _ _ _正如你所看到的，每個都應(yīng)該產(chǎn)生一個 nxn 矩陣。（我可以用 np.NaN 替換每個“_”）我將在下面發(fā)布解決方案。非常感謝所有的貢獻(xiàn)。
3 回答
慕萊塢森
TA貢獻(xiàn)1810條經(jīng)驗 獲得超4個贊
您可以像@user2357112 建議的那樣簡化一些事情：
import numpy as np
import pandas as pd
n = 6
arr = ([i*n+j for i in range(1,n-j+1) if i<=j+1] for j in range(n))
df = pd.DataFrame([np.NaN]+x+[np.NaN]*(n-len(x)-1) for x in arr)
print(df)
輸出：
0 1 2 3 4 5
0 NaN 6 NaN NaN NaN NaN
1 NaN 7 13.0 NaN NaN NaN
2 NaN 8 14.0 20.0 NaN NaN
3 NaN 9 15.0 21.0 NaN NaN
4 NaN 10 16.0 NaN NaN NaN
5 NaN 11 NaN NaN NaN NaN
慕森卡
TA貢獻(xiàn)1806條經(jīng)驗 獲得超8個贊
你可以把它放在內(nèi)部列表理解中：
arr = [[(i*n+j) for i in range(n) if ( 2*i<=j+i+1<=n+1 and i>0)] for j in range(n)]
輸出：
n = 10 # for example
print(arr)
[[10],
[11, 21],
[12, 22, 32],
[13, 23, 33, 43],
[14, 24, 34, 44, 54],
[15, 25, 35, 45, 55],
[16, 26, 36, 46],
[17, 27, 37],
[18, 28],
[19]]
編輯：
如果您希望在具有所需輸出的 DataFrame 中使用它：
import numpy as np
import pandas as pd
n = 10
arr = [[np.NaN] + [(i*n+j) for i in range(n) if ( 2*i<=j+i+1<=n+1 and i>0)] + [np.NaN] * (n - j - 2) for j in range(n)]
pd.DataFrame(arr)
0 1 2 3 4 5 6 7 8 9
0 NaN 10 NaN NaN NaN NaN NaN NaN NaN NaN
1 NaN 11 21.0 NaN NaN NaN NaN NaN NaN NaN
2 NaN 12 22.0 32.0 NaN NaN NaN NaN NaN NaN
3 NaN 13 23.0 33.0 43.0 NaN NaN NaN NaN NaN
4 NaN 14 24.0 34.0 44.0 54.0 NaN NaN NaN NaN
5 NaN 15 25.0 35.0 45.0 55.0 NaN NaN NaN NaN
6 NaN 16 26.0 36.0 46.0 NaN NaN NaN NaN NaN
7 NaN 17 27.0 37.0 NaN NaN NaN NaN NaN NaN
8 NaN 18 28.0 NaN NaN NaN NaN NaN NaN NaN
9 NaN 19 NaN NaN NaN NaN NaN NaN NaN NaN
梵蒂岡之花
TA貢獻(xiàn)1900條經(jīng)驗 獲得超5個贊
from pandas import DataFrame as df
import numpy as np
n = {USERINPUT_Var_(int>1)}
arr = [[i*n+j if ( 2*i<=j+i+1<=n+1 and i>0) else np.NaN for i in range(n)] for j in range(n)]
print(df(arr))
我現(xiàn)在仍然可以使用一些幫助來簡化條件。
添加回答
舉報
0/150
提交
取消