在使用gRPC作為遠程調用框架時,如何使用python來表示多維數組呢?gRPC中定義proto文件時,有一個參數是repeated,用來表示重復的數據類型,使用這個參數可以表示list類型。如下,我想表示一個三維數組,這個數組表示的是一個三通道RGB彩色圖像,proto文件如下所示,proto文件中GrayFeature表示的是一個通道的灰度圖像。
syntax = "proto3";
service DataListServer {
rpc TransDataList(stream GrayFeature) returns (Reply) {}
}
message GrayFeature {
repeated Feature feature = 1;
}
message Feature {
repeated int32 pixel = 2;
}
對應的Python代碼如下:
data_list = ndarray.tolist()
lenth = len(data_list)
for j in range(0, lenth):
matrix = data_list[j]
gray_feature = data_list_pb2.GrayFeature()
for column in range(0, len(matrix)):
feature = gray_feature.feature.add()
feature.pixel.extend(matrix[column])
yield gray_feature
在服務端對應的解析的Python代碼如下:
data_list = [[[]]] * 3
for gray_feature in features:
j = 0
feature = gray_feature.feature
length = len(feature)
matrix = [[]] * length
for i in range(0, length):
matrix[i] = feature[i].pixel
data_list[j] = matrix
j += 1
