0

我正在尝试将 processing.org 应用程序移植到 python 并遇到一些困难。

我需要用python写这个：

int[][] elevation; // elevations in meters
float[][] felevation; // scaled to range [0,1] where 1=max
int maxheight;

void setup(){
size(600,600,P2D);

// read srtm binary file
elevation=new int[1201][1201];
felevation=new float[1201][1201];
byte b[] = loadBytes("N30W091.hgt"); // THIS IS A BINARY FILE
int ix=0;
maxheight=0;
for (int row=0;row<1201;row++) {
  for (int col=0;col<1201;col++) {
    // bytes are signed, from -128 to 127, converts to unsigned...
    int hi = b[ix] & 0xff; 
    int lo = b[ix+1] & 0xff; 
    int el=(int)((hi<<8)|lo); // big endian!
    elevation[row][col]=el;
    if (el>maxheight && el<32000) maxheight=el; 
    ix+=2;
   }
}

... 等等

到目前为止我所做的是：

elevation = [[],[]]
maxheight=0

b = open("C:\\Users\\CNA\\sketchbook\\_SRTM\\data\\N59E010.hgt","rb")
fin = b.read(1)
print(len(fin))
ix = 0
for row in range(0,1201):
    for col in range(0,1201):
        hi = (fin[ix]   + 0xff)
        lo = (fin[ix+1] + 0xff)

我总是得到

Traceback (most recent call last):

  File "C:\Users\CNA\workspace\Revitter\PatternAsignment.py", line 16, in <module>

TypeError: unsupported operand type(s) for +: 'str' and 'int'

有任何想法吗？..我是python新手，我没有使用字节的经验...

4

3 回答 3

5

惯用的翻译将以完全不同的方式工作。

在原始代码中，您进行了一堆位旋转以将两个字节值转换为单个数值。在 Python 中，为此提供了内置功能：使用struct模块。事实证明，这个模块已经为一次读取多个值而构建。

此外，对文件名使用正斜杠 - 它更容易，并且可以保证工作。使用 with-block 确保文件自动正确关闭，并使用列表推导来简化循环 - 不要试图告诉 Python 如何构建列表，而只需询问您想要的列表。

这给了我们：

import struct
with open('C:/Users/CNA/sketchbook/_SRTM/data/N59E010.hgt', 'rb') as data:
    elevation = [
        list(struct.unpack('>1201H', data.read(1201 * 2)))
        for row in range(1201)
    ]
maxheight = max(max(cell for cell in row if cell < 32000) for row in elevation)

你就完成了。欢迎来到 Python :)

于 2012-05-05T14:25:39.990 回答
2

在 Python 中，like 值'hello'[2]也是字符串（在这种情况下 == 'l'）。您需要使用将它们转换为整数ord并使用chr.

elevation = [[],[]]
maxheight=0

b = open("C:\\Users\\CNA\\sketchbook\\_SRTM\\data\\N59E010.hgt","rb")
fin = b.read() # you probably need to read more than 1 byte, this will read whole file
print(len(fin))
ix = 0
for row in range(0,1201):
    for col in range(0,1201):
        hi = (ord(fin[ix])   + 0xff) # ord returns unsigned integer, so you probably don't need to convert it
        lo = (ord(fin[ix+1]) + 0xff)
        el = (hi << 8) | lo

请参阅：http ://docs.python.org/library/functions.html

于 2012-05-05T13:20:11.150 回答
0

我不确定我的 .reshape 语法是否正确（将进一步测试） - 但非常像这样的东西应该可以满足您的需求：

import numpy

def readHgtFile(fname, h=1201, w=1201):
    with open(fname, 'rb') as inf:
        return numpy.fromfile(inf, dtype=[('height', '>u2')], count=h*w).reshape((h,w))

def main():
    elevation = readHgtFile('N30W091.hgt')
    maxheight = elevation.max()

if __name__=="__main__":
    main()
于 2012-05-05T15:00:31.787 回答